Artificial Intelligence Engineer - PyTorch/Tensorflow

About the Role :

We are looking for an AI Engineer with 3+ years of strong hands-on experience in designing, developing, and deploying machine learning and Generative AI solutions. The ideal candidate is highly capable in end-to-end model development, LLM fine-tuning, and AI agent automation, with a proven ability to build production-ready AI systems.

Key Responsibilities :

AI/ML Model Development :

- Apply optimization techniques such as quantization, pruning, distillation, and parameter-efficient fine tuning (LoRA, Adapters).

- Perform large-scale training and distributed learning where needed.

Generative AI & LLM Engineering :

- Apply prompt engineering, domain adaptation, and advanced fine-tuning techniques.

AI Agent & Workflow Automation :

- Maintain reproducible and observable pipelines using MLflow, Airflow, LangSmith, etc.

Model Deployment & MLOps :

- Build scalable APIs using FastAPI with monitoring, logging, and load balancing.

- Implement CI/CD pipelines for automated ML deployments.

Monitoring, Optimization & Reliability :

- Optimize models for production workloads and reduce latency/resource usage.

Collaboration :

- Contribute to architectural decisions and documentation.

Required Qualifications :

Core AI/ML Expertise :

- Strong hands-on experience with LLMs, fine-tuning methods (LoRA/Adapters), and GenAI workloads.

- Experience deploying ML models at scale with autoscaling, monitoring, and performance optimization.

Technical Skills :

- Strong understanding of ML pipelines, APIs, Git, unit testing, and software engineering best practices.

- Hands-on experience with Docker, Kubernetes, and cloud ML platforms (AWS/Azure/GCP).

- Experience with MLflow, Airflow, LangSmith, Weights & Biases, or similar tools.

AI Agent & GenAI Development :

- Experience building production-ready GenAI-based assistants.

Preferred Qualifications :

- Knowledge of model compression and edge deployment techniques.

- Understanding of distributed training, privacy-preserving ML, or federated learning.

- Research publications or contributions to open-source AI projects.

Education :

- Masters degree or cloud ML certifications (AWS/GCP/Azure) are a plus.

What We Offer :

- Access to high-performance compute and modern ML tooling.

- A collaborative, innovation-led environment.

- Competitive salary, benefits, and growth opportunities.