HubMub
-
Arcitech

Python AI/ML Developer

Arcitech
India · Full-time · Mid-Senior
Python Developer - AI/MLYour Responsibilities

  • Develop, train, and optimize ML models using PyTorch, TensorFlow, and Keras.
  • Build end-to-end LLM and RAG pipelines using LangChain and LangGraph.
  • Work with LLM APIs (OpenAI, Anthropic Claude, Azure OpenAI) and implement prompt engineering strategies.
  • Utilize Hugging Face Transformers for model fine-tuning and deployment.
  • Integrate embedding models for semantic search and retrieval systems.
  • Work with transformer-based architectures (BERT, GPT, LLaMA, Mistral) for production use cases.
  • Implement LLM evaluation frameworks (RAGAS, LangSmith) and performance optimization.
  • Design and maintain Python microservices using FastAPI with REST/GraphQL APIs.
  • Implement real-time communication with FastAPI WebSockets.
  • Implement pgvector for embedding storage and similarity search with efficient indexing strategies.
  • Integrate vector databases (pgvector, Pinecone, Weaviate, FAISS, Milvus) for retrieval pipelines.
  • Containerize AI services with Docker and deploy on Kubernetes (EKS/GKE/AKS).
  • Configure AWS infrastructure (EC2, S3, RDS, SageMaker, Lambda, CloudWatch) for AI/ML workloads.
  • Version ML experiments using MLflow, Weights & Biases, or Neptune.
  • Deploy models using serving frameworks (TorchServe, BentoML, TensorFlow Serving).
  • Implement model monitoring, drift detection, and automated retraining pipelines.
  • Build CI/CD pipelines for automated testing and deployment with ≥80% test coverage (pytest).
  • Follow security best practices for AI systems (prompt injection prevention, data privacy, API key management).
  • Participate in code reviews, tech talks, and AI learning sessions.
  • Follow Agile/Scrum methodologies and Git best practices.

Required Qualifications

  • Bachelor's or Master's degree in Computer Science, AI/ML, or related field.
  • 2–5 years of Python development experience (Python 3.9+) with strong AI/ML background.
  • Hands-on experience with LangChain and LangGraph for building LLM-powered workflows and RAG systems.
  • Deep learning experience with PyTorch or TensorFlow.
  • Experience with Hugging Face Transformers and model fine-tuning.
  • Proficiency with LLM APIs (OpenAI, Anthropic, Azure OpenAI) and prompt engineering.
  • Strong experience with FastAPI frameworks.
  • Proficiency in PostgreSQL with pgvector extension for embedding storage and similarity search.
  • Experience with vector databases (pgvector, Pinecone, Weaviate, FAISS, or Milvus).
  • Experience with model versioning tools (MLflow, Weights & Biases, or Neptune).
  • Hands-on with Docker, Kubernetes basics, and AWS cloud services.
  • Skilled in Git workflows, automated testing (pytest), and CI/CD practices.
  • Understanding of security principles for AI systems.
  • Excellent communication and analytical thinking.

Nice to Have

  • Experience with multiple vector databases (Pinecone, Weaviate, FAISS, Milvus).
  • Knowledge of advanced LLM fine-tuning (LoRA, QLoRA, PEFT) and RLHF.
  • Experience with model serving frameworks and distributed training.
  • Familiarity with workflow orchestration tools (Airflow, Prefect, Dagster).
  • Knowledge of quantization and model compression techniques.
  • Experience with infrastructure as code (Terraform, CloudFormation).
  • Familiarity with data versioning tools (DVC) and AutoML.
  • Experience with Streamlit or Gradio for ML demos.
  • Background in statistics, optimization, or applied mathematics.
  • Contributions to AI/ML or LangChain/LangGraph open-source projects.

Skills:- Large Language Models (LLM), Retrieval Augmented Generation (RAG), PyTorch, Keras, pytest, WebSocket, SQL, TensorFlow and langraph

Key Skills

Ranked by relevance

python ai tensorflow fastapi pytorch kubernetes storage docker mlflow cicd git aws infrastructure as code microservices postgresql terraform streamlit cloud keras sql s3
Login to Apply
Posted
Nov 19, 2025
Type
Full-time
Level
Mid-Senior
Location
Navi Mumbai
Company
Arcitech

Industries

Software Development

Categories

Engineering Information Technology

Related Jobs

3 roles aligned with this opportunity

View all jobs
View Job Details
XLSCOUT
Related

Data Scientist

2026-06-17

Full-time
Mid-Senior
India
Software Development
Product Management
View Job Details
Scrumconnect Consulting
Related

Graduate AI Engineer

2026-06-15

Volunteer
Not Applicable
United Kingdom
Software Development
Engineering
View Job Details
HP
Related

Machine Learning Engineer - AI Applications

2026-06-15

Full-time
Mid-Senior
Brazil
Computer Hardware Manufacturing
Engineering