AI Engineer (FDE)

About the role

You will also work on problems where existing architectures may not be sufficient and will be expected to experiment with new approaches that combine large language models, machine learning models, and data engineering patterns to build reliable, production-grade systems Location - Mumbai/Bangalore/Hyderabad/Gurgaon (Hybrid - 3 Days a week in Office)

What you'll do

• Design and architect modular AI agent frameworks incorporating skill decomposition, tool orchestration, and persistent state tracking
• Build and deploy supervised and unsupervised ML models for prediction, classification, anomaly detection, and pattern recognition tasks in production environments
• Develop decision-making loops that balance trade-offs between exploration vs. exploitation, cost vs. accuracy, and latency vs. reasoning depth
• Build structured memory systems including episodic memory stores, semantic memory layers, and vector-based memory with optimised retrieval strategies
• Design tool-calling architectures with strong execution validation, retry mechanisms, and failure recovery strategies

What you need

• 5–9 yrs of experience
• Machine learning
• Scikit-learn
• Python
• Prompt engineering
• PyTorch
• Databricks
• Any graduate

Good to have

• GCP
• MLOps
• Azure
• Terraform
• Grafana
• Prometheus