Python Developer (AI/ML)

Job Description

Designation: Python Developer (Experienced in AI/ML) Location: Turbhe, Navi Mumbai CTC: 6-12 LPA Years of Experience: 2-5 years 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. 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 ??