Level 2

Python Developer (AI/ML)

Navi Mumbai
Full time
₹ 6 LPA to 12 LPA

Job Description

Location: Turbhe, Navi Mumbai
Budget: ₹ 6 LPA to 12 LPA
Experience Required: 2 - 5 years
Availability: Immediate Joiners Preferred

Why Join Us?
- Work on cutting-edge LLM and AI/ML applications
- Competitive compensation & benefits
- Supportive, collaborative & innovation-first culture

Role Overview
We are looking for a Python Developer (AI/ML) who is passionate about building scalable, cloud-native platforms and next-gen intelligent products. You’ll work with state-of-the-art AI/ML frameworks, deploy LLM-powered systems into production, and collaborate with cross-functional teams to deliver innovative AI-driven solutions.

Key Responsibilities

AI/ML Engineering
- Develop, train, and optimize ML models using PyTorch/TensorFlow/Keras.
- Build end-to-end LLM and RAG pipelines with LangChain.
- Collaborate with data scientists to productionize prototypes.
- Integrate NLP, Computer Vision, and Recommendation Systems into products.
- Apply transformer-based models (BERT, GPT, LLaMA, etc.) for real-world use cases.

Backend & Systems Development
- Design and maintain Python microservices with REST/GraphQL APIs.
- Implement real-time communication using Django Channels/WebSockets.
- Containerize and deploy AI services with Docker & Kubernetes.
- Configure and manage AWS (EC2, S3, RDS, SageMaker, CloudWatch) for AI workloads.

Reliability & Automation
- Build background task queues with Celery for reliable execution.
- Implement CI/CD pipelines for model training, testing, and deployment.
- Write automated unit & integration tests (pytest/unittest) with ≥80% coverage.

Collaboration
- Contribute to MLOps best practices & mentor peers in LangChain/AI integration.
- Participate in tech talks, code reviews, and AI learning sessions.

Key Skills We’re Looking For
- Strong Python development expertise
- Hands-on with LangChain and RAG pipeline design
- Experience with PyTorch/TensorFlow/Keras
- Knowledge of REST/GraphQL API development
- Cloud expertise (AWS/Azure/GCP)
- Familiarity with Docker, Kubernetes, and CI/CD workflows
- Git-based workflows, automated testing practices

Nice to Have
- Experience with vector databases (Pinecone, FAISS, Weaviate, Milvus)
- LLM fine-tuning, prompt engineering, evaluation frameworks
- Familiarity with Airflow/Prefect/Dagster for pipelines
- Background in statistics, optimization, or applied mathematics
- Open-source contributions in AI/ML or LangChain
- Model monitoring & drift detection in production

You have successfully applied for the position

We will get back to you on email