Senior AI/ML Developer / Lead AI/ML Developer

About Us Amantra is an AI-driven technology company focused on building intelligent, scalable, and enterprise-ready solutions powered by Agentic AI, automation, and advanced technologies. We are looking for an experienced Senior AI/ML Developer or Lead AI/ML Developer to drive the design, development, and deployment of AI/ML solutions and contribute to the development of next-generation intelligent applications.

Responsibilities

• Design, develop, and deploy scalable AI/ML models and intelligent applications for enterprise use cases.
• Lead the development of Generative AI and Agentic AI solutions, including AI agents, LLM-powered applications, and multi-agent workflows.
• Develop and optimize machine learning and deep learning models for classification, prediction, natural language processing, and other relevant use cases.
• Work with Large Language Models (LLMs), Retrieval-Augmented Generation (RAG), vector databases, embeddings, and prompt engineering.
• Design and implement AI pipelines, data preprocessing workflows, model evaluation frameworks, and inference services.
• Build and integrate AI solutions using Python, REST APIs, and relevant AI/ML frameworks.
• Evaluate and implement suitable models, libraries, and AI platforms based on business requirements.
• Optimize model performance, latency, scalability, reliability, and inference costs.
• Establish best practices for model versioning, deployment, monitoring, testing, and continuous improvement.
• Collaborate with product managers, software engineers, and other stakeholders to translate business requirements into technical solutions.
• Lead technical discussions, conduct code reviews, mentor junior developers, and promote engineering best practices.
• Stay updated on emerging developments in Generative AI, Agentic AI, and machine learning technologies.

Requirements

• 5+ years of hands-on experience in AI/ML development, with demonstrated experience delivering production-ready solutions.
• Strong proficiency in Python and commonly used libraries such as NumPy, Pandas, Scikit-learn, and relevant deep learning frameworks.
• Strong understanding of machine learning algorithms, deep learning architectures, statistics, and model evaluation techniques.
• Practical experience with Generative AI, LLMs, prompt engineering, embeddings, and RAG-based applications.
• Experience with AI agent frameworks and orchestration tools such as LangChain, LangGraph, LlamaIndex, or equivalent technologies.
• Experience with frameworks such as PyTorch or TensorFlow.
• Hands-on experience with vector databases such as FAISS, Pinecone, Weaviate, Chroma, or equivalent solutions.
• Experience building APIs and integrating AI models into enterprise applications.
• Knowledge of Docker, Git, CI/CD, and cloud-based AI/ML deployment environments.
• Experience with model monitoring, performance optimization, and production troubleshooting.
• Strong problem-solving, analytical, communication, and technical decision-making skills.

Preferred Qualifications

• Experience in developing and deploying Agentic AI projects or products, including AI agents, multi-agent systems, autonomous workflows, and LLM-powered enterprise applications, will be strongly preferred.
• Experience leading AI/ML projects or mentoring developers.
• Exposure to AWS, Azure, or Google Cloud AI services.
• Experience with MLOps, MLflow, model registries, and automated model deployment pipelines.
• Understanding of AI security, responsible AI practices, data privacy, and LLM evaluation.
• Experience developing enterprise-grade AI products, intelligent automation solutions, or Agentic AI applications.

How to Apply

Ready to take the next step in your career? Apply now and become part of our growing team. You can also send your resume to [email protected]