Junior Python Developer

Arcitech AI, located in Mumbai's bustling Lower Parel, is a trailblazer in software and IT, specializing in software development, AI, mobile apps, and integrative solutions. Committed to excellence and innovation, Arcitech AI offers incredible growth opportunities for team members. Enjoy unique perks like weekends off and a provident fund. Our vibrant culture is friendly and cooperative, fostering a dynamic work environment that inspires creativity and forward-thinking. Join us to shape the future of technology.

Key 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 solutions.
• Work with transformer-based architectures (BERT, GPT, LLaMA, etc.) for real-world use cases.

Backend & Systems Development

• Design and maintain robust Python microservices with REST/GraphQL APIs.
• Implement real-time communication with Django Channels/WebSockets.
• Containerize AI services using 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.
• Build 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 AI/LLM integration.
• Participate in tech talks, code reviews, and AI learning sessions within the team.

Required Qualifications

• Bachelor’s/Master’s degree in Computer Science, AI/ML, or related field.
• 8 months to 2yrs of experience in Python development with strong AI/ML exposure.
• Hands-on experience with LangChain for building LLM-powered workflows and RAG systems.
• Strong deep learning expertise with PyTorch/TensorFlow.
• Proven experience deploying ML models and LLM apps into production.
• Familiarity with REST/GraphQL APIs and cloud platforms (AWS/Azure/GCP).
• Proficient in Git workflows, automated testing, and CI/CD practices.

Nice to Have

• Experience with vector databases (Pinecone, Weaviate, FAISS, Milvus).
• Knowledge of LLM fine-tuning, prompt engineering, and evaluation frameworks.
• Familiarity with Airflow/Prefect/Dagster for ML pipelines.
• Background in statistics, optimization, or applied mathematics.
• Contributions to open-source AI/ML or LangChain projects.
• Experience with model monitoring and drift detection.

Why Join Us?

• Work on cutting-edge LLM and AI/ML applications
• A collaborative, innovation-driven work culture
• Opportunities to grow into AI/ML leadership roles