Python Developer AI/ML

Location: Onsite, Navi Mumbai

At Arcitech.ai, we’re redefining the future with AI-powered software solutions across education, recruitment, marketplaces, and beyond. We’re looking for a Python Developer passionate about AI/ML, who’s ready to work on scalable, cloud-native platforms and help build the next generation of intelligent, LLM-driven products.

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.
• Strong proficiency with NumPy, Pandas, Scikit-learn.
• 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

Job Type: Full-time

Pay: ₹600,000.00 - ₹1,400,000.00 per year

Benefits:

• Provident Fund

Application Question(s):

• Will you be able to commute to Vashi, Navi Mumbai?
• Your current CTC?
• Your years of experience?

Work Location: In person