Python Developer (AI/ML)

Designation: Python Developer (Experienced in AI/ML)

Location: Turbhe, Navi Mumbai

CTC: 6-12 LPA

Years of Experience: 2-5 years

Arcitech.ai

Your Responsibilities

AI/ML Engineering

• Develop, train, and optimize ML models using

  PyTorch/TensorFlow/Keras

  .
• Build

  end-to-end LLM and RAG (Retrieval-Augmented Generation) pipelines

  using

  LangChain

  .
• Collaborate with data scientists to convert prototypes into

  production-grade AI applications

  .
• Integrate

  NLP, Computer Vision, and Recommendation Systems

  into scalable products.
• Work with

  transformer-based architectures (BERT, GPT, LLaMA, etc.)

  for real-world AI use cases.

Backend & Systems Development

• Design, develop, and maintain robust

  Python microservices

  with

  REST/GraphQL APIs

  .
• Implement

  real-time communication

  with Django Channels/WebSockets.
• Containerize AI services with

  Docker

  and deploy on

  Kubernetes (EKS/GKE/AKS)

  .
• Configure and manage

  AWS (EC2, S3, RDS, SageMaker, CloudWatch)

  for AI/ML workloads.

Reliability & Automation

• Develop background task queues with

  Celery

  , ensuring smart retries and monitoring.
• Implement

  CI/CD pipelines

  for automated model training, testing, and deployment.
• Write

  automated unit & integration tests

  (pytest/unittest) with ≥80% coverage.

Collaboration

• Contribute to

  MLOps best practices

  and mentor peers in

  LangChain/AI integration

  .
• Participate in

  tech talks, code reviews, and AI learning sessions

  within the team.

Required Qualifications

• Bachelor’s or Master’s degree in

  Computer Science, AI/ML, or related field

  .
• 2–5 years of experience in

  Python development with strong AI/ML exposure

  .
• Hands-on experience with

  LangChain

  for building LLM-powered workflows and RAG systems.
• Deep learning experience with

  PyTorch or TensorFlow

  .
• Experience deploying ML models and LLM apps into

  production systems

  .
• Familiarity with

  REST/GraphQL APIs

  and cloud platforms (AWS/Azure/GCP).
• Skilled in

  Git workflows, automated testing, and CI/CD practices

  .

Nice to Have

• Experience with

  vector databases (Pinecone, Weaviate, FAISS, Milvus)

  for retrieval pipelines.
• Knowledge of

  LLM fine-tuning, prompt engineering, and evaluation frameworks

  .
• Familiarity with

  Airflow/Prefect/Dagster

  for data and model pipelines.
• Background in

  statistics, optimization, or applied mathematics

  .
• Contributions to

  AI/ML or LangChain open-source projects

  .
• Experience with

  model monitoring and drift detection

  in production.

Why Join Us

• Competitive compensation and benefits 💰
• Work on

  cutting-edge LLM and AI/ML applications

  🤖
• A collaborative, innovation-driven work culture 📚
• Opportunities to

  grow into AI/ML leadership roles

  🚀