AI/ML Developer (Immediate joiner)

Job Title: AI/ML Developer

Location: Delhi
Experience: 3 years

Apply here - https://beyondroot.keka.com/careers/jobdetails/36292

We are looking for a highly skilled and motivated AI/ML Developer to join our growing team. You'll work on cutting-edge projects involving Large Language Models (LLMs), deep learning, NLP, and model deployment. This is a hands-on role for someone passionate about solving real-world problems with AI.

Key Responsibilities

• Develop and fine-tune deep learning models using PyTorch or TensorFlow
• Build and optimize LLM applications using Hugging Face Transformers and LangChain
• Design prompt engineering strategies and perform fine-tuning for specific use cases
• Implement NLP pipelines (tokenization, embeddings, attention mechanisms, etc.)
• Work with vector databases (FAISS, Pinecone, Weaviate) for semantic search and RAG
• Build scalable APIs using FastAPI or Flask
• Deploy models using Docker, ONNX, TorchScript, and REST APIs
• Collaborate with DevOps to deploy solutions on cloud platforms (AWS, GCP, or Azure)
• Maintain version control using Git and GitHub

Required Technical Skills

• Python programming
• Deep Learning frameworks: PyTorch, TensorFlow
• Transformers & LLMs: Hugging Face Transformers, LangChain
• Prompt Engineering & Fine-tuning
• NLP fundamentals: tokenization, embeddings, attention mechanisms
• Data preprocessing: NumPy, Pandas, and OpenCV (for image data)
• Vector databases: FAISS, Pinecone, Chroma, Weaviate
• API development: FastAPI, Flask
• Model deployment: Docker, REST APIs, ONNX, TorchScript
• Version control: Git, GitHub
• Cloud Platforms: AWS / GCP / Azure

Nice to Have / Advanced Skills

• Retrieval-Augmented Generation (RAG)
• LLM Orchestration: LangGraph, LlamaIndex
• Knowledge Graphs & Semantic Search
• MLOps tools: MLflow, DVC, Kubeflow
• Distributed training: DeepSpeed, Hugging Face Accelerate
• Vector search optimization: Milvus, Qdrant
• Custom model training: LoRA, PEFT, RLHF
• Front-end: React, Streamlit, Gradio
• Databases: PostgreSQL, MongoDB
• Monitoring & logging: Prometheus, Grafana
• Prompt security / Data governance

Job Types: Full-time, Permanent

Pay: ₹40,000.00 - ₹90,000.00 per month

Work Location: In person