Senior ML Engineer at Pi · Mumbai · 4 - 7 years · Raised funding · Posted 1 Jun 2024

About us:
Pi - India’s 1st SuperMoney App, is an AI-native Fintech startup building ML platforms for financial advisory and wealth management. Founded by a former Investment Banker (IIT/IIM/ CFA®) with over a decade of experience.
We are looking for Machine Learning Engineers who are passionate about developing world-class ML platforms. You will be part of the Core Founding team that lays the foundation for a Unicorn journey. You'll join a visionary, funded startup on the ground floor and get to build it up one conversation at a time.
Qualifications:
- Overall experience of 4 to 7 years in ML
- B.Tech/BE from top-tier colleges (IIT, NIT, IIIT)
Requirements:
- Applied GenAI expertise in implementing cutting-edge LLMs and diverse generative models, specializing in transformers and GPT models.
- Demonstrated proficiency across various modalities, showcasing a dynamic skill set for innovative solutions.
- Expertise in Natural Language Processing (NLP), with a comprehensive understanding of the entire stack involved in designing, training, evaluating, and deploying machine learning models, particularly large language models.
- Deep understanding of the entire stack when it comes to designing, training, evaluating, and deploying ML models. Successfully collected a new giant dataset and played a critical role in shipping a new ML product that required custom components.
- Extensive experience in designing and building intelligent agents, leveraging AI techniques to create autonomous and adaptive systems for various applications.
- Proficient in writing distributed ML infrastructure, debugging and fixing hard-to-find bugs in ML models, and owning production-grade projects from start to finish.
- Experience with Deep learning frameworks such as PyTorch, TensorFlow. , Jax, and Keras.
- Proficient in writing GPU kernels using CUDA .
- Experience with AWS SageMaker and the AWS Machine Learning suite of tools.
- Developed novel techniques for ML model measurement and mitigation.
- Proficient in Apache Kafka for seamless data streaming, Apache Spark for large-scale batch processing, and Apache Airflow for efficient orchestration.
- Demonstrated expertise in Kubernetes-based workflows, specifically Kubeflow, for scalable and portable machine learning and data processing deployments, enhancing pipeline reliability and performance.
- Experience with MongoDB, MySQL, and Vector databases.
- Experience with analytics.
- Proficient in Kubernetes, Python, Github, Huggingface, R(for Analytics)
- Experience in mentoring and growing engineers.
Additional Benefits:
Competitive compensation along with ESOPs of a path-breaking Fintech startup

About Pi
About
Company social profiles
Similar jobs (10)
Strong Senior AI Engineer Profile with production-grade LLM and Python expertise
2
Mandatory (Experience 1): Must have at least 3+ years of professional AI/ML engineering experience with demonstrated track record of delivering production-grade AI systems in real-world environments
3
Mandatory (Tech skill 1): Must have strong programming skills in Python and SQL, with hands-on ML/data libraries (scikit-learn, pandas, numpy) and deep-learning frameworks (PyTorch or TensorFlow), plus REST API design
4
Mandatory (Experience 2): Must have hands-on experience building and deploying production-grade ML/LLMs including RAG pipelines, document parsing, information extraction, and text processing on large-scale unstructured data (preprocessing, chunking, embeddings, feature engineering)
5
Mandatory (Tech skill 2): Must have strong NLP / extraction-focused ML depth — transformers, embeddings, vector databases, RAG, LLM integrations, and agentic workflows.
6
Mandatory (Tech skill 3): Must have hands-on experience with AWS (SageMaker, Bedrock, EC2, Lambda) and Azure (AI Foundry, Azure OpenAI) for model training, fine-tuning, deployment, and inference at scale
7
Mandatory (Tech skill 4): Must have experience with multi-agentic frameworks / orchestration tools (Claude Code, LangGraph, LangChain, CrewAI)
8
Mandatory (Tech skill 5): Must have hands-on experience with MLOps ecosystem including experiment tracking (MLflow, Weights & Biases), model versioning/registry, automated retraining, and CI/CD (GitHub Actions, Azure DevOps, Docker, Kubernetes)
9
Mandatory (Tech skill 6): Must have experience with evaluation frameworks (precision, recall, F1, field-level accuracy) and iterative model improvement, plus AI observability (Prometheus, Grafana, SLOs).
10
Mandatory (Team Leadership): Must have experience leading projects or teams, managing technical deliverables, and mentoring, with strong problem-solving in ambiguous challenges
11
Mandatory (Education) - B.Tech or B.E. If graduated in BCA or BCS, they should have a Maters or Post graduation
12
Mandatory (Company): B2B Tech/Consulting/B2B Software
Kody Technolab Limited is seeking an experienced AI/ML Engineer to design, develop, and deploy cutting-edge Artificial Intelligence and Machine Learning solutions. The ideal candidate will have
strong expertise in Machine Learning, Deep Learning, Generative AI, LLMs, MLOps, and cloud-based AI deployments.
Key Responsibilities
• Design, develop, and deploy Machine Learning and Deep Learning models for classification, regression, recommendation systems, NLP, Computer Vision, and Generative AI applications.
• Build and maintain end-to-end ML pipelines including data preprocessing, feature engineering, model training, validation, evaluation, and deployment.
• Develop AI solutions using PyTorch, TensorFlow, Scikit-learn, Hugging Face, and related frameworks.
• Work with Large Language Models (LLMs) and foundation models such as GPT, BERT, Llama, Claude, and Stable Diffusion.
• Collaborate with product, engineering, and business teams to translate requirements into scalable AI solutions.
• Optimize model performance, scalability, and reliability for production environments.
• Implement MLOps best practices using tools such as MLflow, Docker, Kubernetes, and Kubeflow.
• Stay updated with emerging trends and research in AI, ML, Deep Learning, and Generative AI.
Required Qualifications
• Bachelor’s or Master’s degree in Computer Science, Data Science, Artificial Intelligence, Mathematics, or a related field.
• 7+ years of hands-on experience in AI/ML product development.
• Strong proficiency in Python and ML frameworks including Scikit-learn, TensorFlow, PyTorch, and Hugging Face.
• Experience with Generative AI, LLMs, GANs, VAEs, diffusion models, and prompt engineering.
• Strong understanding of the ML lifecycle including model training, tuning, deployment, monitoring, and optimization.
• Experience with MLOps tools such as MLflow, Docker, Kubeflow, and CI/CD pipelines.
• Experience with AWS, Azure, or GCP cloud platforms.
• Strong problem-solving and analytical skills.
Preferred Skills
• Fine-tuning and deployment of Large Language Models.
• Experience with RAG (Retrieval Augmented Generation) architectures.
• Contributions to open-source AI projects or research publications.
• Knowledge of model interpretability, data annotation, and feature engineering.
• C++ experience for high-performance AI applications.
Why Join Kody Technolab Limited?
Opportunity to work on innovative AI products, Generative AI solutions, robotics integrations,
and enterprise-scale applications while collaborating with a highly skilled technology team.
Visit the Website to know more about us.
Company Website - Kody Technolab | Deep Tech Company in Robotics & AI Solution
Kody Robots | Robotics Company in India for Autonomous Robots
Kody Technolab Limited
AI/ML Lead (7+ Years Experience)
Location: Ahmedabad / Gandhinagar
Experience: 7+ Years
Employment Type: Full-Time
Kody Technolab Limited is seeking an experienced AI/ML Engineer to design, develop, and deploy
cutting-edge Artificial Intelligence and Machine Learning solutions. The ideal candidate will have
strong expertise in Machine Learning, Deep Learning, Generative AI, LLMs, MLOps, and
cloud-based AI deployments.
Key Responsibilities
• Design, develop, and deploy Machine Learning and Deep Learning models for classification,
regression, recommendation systems, NLP, Computer Vision, and Generative AI applications.
• Build and maintain end-to-end ML pipelines including data preprocessing, feature
engineering, model training, validation, evaluation, and deployment.
• Develop AI solutions using PyTorch, TensorFlow, Scikit-learn, Hugging Face, and related
frameworks.
• Work with Large Language Models (LLMs) and foundation models such as GPT, BERT, Llama,
Claude, and Stable Diffusion.
• Collaborate with product, engineering, and business teams to translate requirements into
scalable AI solutions.
• Optimize model performance, scalability, and reliability for production environments.
• Implement MLOps best practices using tools such as MLflow, Docker, Kubernetes, and Kubeflow.
• Stay updated with emerging trends and research in AI, ML, Deep Learning, and Generative AI.
Required Qualifications
• Bachelor’s or Master’s degree in Computer Science, Data Science, Artificial Intelligence,
Mathematics, or a related field.
• 7+ years of hands-on experience in AI/ML product development.
• Strong proficiency in Python and ML frameworks including Scikit-learn, TensorFlow, PyTorch, and
Hugging Face.
• Experience with Generative AI, LLMs, GANs, VAEs, diffusion models, and prompt engineering.
• Strong understanding of the ML lifecycle including model training, tuning, deployment,
monitoring, and optimization.
• Experience with MLOps tools such as MLflow, Docker, Kubeflow, and CI/CD pipelines.
• Experience with AWS, Azure, or GCP cloud platforms.
• Strong problem-solving and analytical skills.Preferred Skills
• Fine-tuning and deployment of Large Language Models.
• Experience with RAG (Retrieval Augmented Generation) architectures.
• Contributions to open-source AI projects or research publications.
• Knowledge of model interpretability, data annotation, and feature engineering.
• C++ experience for high-performance AI applications.
Why Join Kody Technolab Limited?
Opportunity to work on innovative AI products, Generative AI solutions, robotics integrations,
and enterprise-scale applications while collaborating with a highly skilled technology team.
Visit the Website to know more about us.
Company Website - Kody Technolab | Deep Tech Company in Robotics & AI Solution
Kody Robots | Robotics Company in India for Autonomous Robots
🚀 WE’RE HIRING | AI/ML GENERATIVE AI ENGINEER
📍 Location: Remote
💼 Experience: 5+ Years
🔄 Position: Contract – Extendable
🔹 ROLE HIGHLIGHTS
➤ Build and deploy AI/ML amp; Generative AI solutions
➤ Develop LLM, RAG, NLP, Recommendation amp; Predictive solutions
➤ Work on AI Agents, Chatbots, Computer Vision amp; Content Intelligence
➤ Build ML models using PyTorch, TensorFlow, Keras amp; Scikit-learn
➤ Develop RAG solutions using LangChain, LlamaIndex, FAISS/Milvus
➤ Integrate OpenAI, Azure OpenAI, AWS Bedrock, Vertex AI amp; Hugging Face
➤ Build scalable AI APIs using Python, FastAPI/Flask/Django
➤ Contribute to Private AI amp; Smart Agentic Systems
⚙️ MUST-HAVE SKILLS
◆ Python – 3+ years
◆ AI/ML – 5+ years
◆ Generative AI – 2+ years
◆ LLMs, RAG, Embeddings , Transformers
◆ ML/DL, NLP amp; Predictive Analytics
◆ Cloud AI Platforms – Azure / AWS / GCP
◆ AI/ML Deployment | MLOps
Interview Process - F2F Round at Pune Location
Strong AI Engineer / Machine Learning Engineer profiles.
2
Mandatory (Experience 1) – Must have minimum 3+ years of hands-on experience in Data Science, Machine Learning, Applied AI, NLP, Deep Learning, or Generative AI solutions.
3
Mandatory (Experience 2) – Must have strong hands-on experience in Python programming, SQL, data analysis, feature engineering, model development, and production-grade ML applications.
4
Mandatory (Experience 3) – Must have experience working with Machine Learning and Deep Learning frameworks such as PyTorch, TensorFlow, Keras, Scikit-learn, or equivalent.
5
Mandatory (Experience 4) – Must have hands-on experience working on NLP, embeddings, semantic search, text classification, document understanding, recommendation systems, or similar AI/ML use cases.
6
Mandatory (Experience 5) – Must have experience working with Large Language Models (LLMs) such as GPT, Llama, Mistral, Claude, Gemini, Phi, or similar foundation models.
7
Mandatory (Experience 6) – Must have hands-on experience building or implementing RAG (Retrieval Augmented Generation) systems, vector search, knowledge retrieval, embeddings, chunking, indexing, or semantic retrieval solutions.
8
Mandatory (Experience 7) – Must have experience working with Git, CI/CD practices, production environments, and scalable AI/ML systems.
Mandatory ( Pedigree) - B.TECH / M.TECH from Tier 1 Colleges (IIT's, NIT's, BITS) are Considered.
Strong Senior AI Engineer Profile with production-grade LLM and Python expertise
2
Mandatory (Experience 1): Must have at least 3+ years of professional AI/ML engineering experience with demonstrated track record of delivering production-grade AI systems in real-world environments
3
Mandatory (Tech skill 1): Must have strong programming skills in Python and SQL, with hands-on ML/data libraries (scikit-learn, pandas, numpy) and deep-learning frameworks (PyTorch or TensorFlow), plus REST API design
4
Mandatory (Experience 2): Must have hands-on experience building and deploying production-grade ML/LLMs including RAG pipelines, document parsing, information extraction, and text processing on large-scale unstructured data (preprocessing, chunking, embeddings, feature engineering)
5
Mandatory (Tech skill 2): Must have strong NLP / extraction-focused ML depth — transformers, embeddings, vector databases, RAG, LLM integrations, and agentic workflows.
6
Mandatory (Tech skill 3): Must have hands-on experience with AWS (SageMaker, Bedrock, EC2, Lambda) and Azure (AI Foundry, Azure OpenAI) for model training, fine-tuning, deployment, and inference at scale
7
Mandatory (Tech skill 4): Must have experience with multi-agentic frameworks / orchestration tools (Claude Code, LangGraph, LangChain, CrewAI)
8
Mandatory (Tech skill 5): Must have hands-on experience with MLOps ecosystem including experiment tracking (MLflow, Weights & Biases), model versioning/registry, automated retraining, and CI/CD (GitHub Actions, Azure DevOps, Docker, Kubernetes)
9
Mandatory (Tech skill 6): Must have experience with evaluation frameworks (precision, recall, F1, field-level accuracy) and iterative model improvement, plus AI observability (Prometheus, Grafana, SLOs).
10
Mandatory (Team Leadership): Must have experience leading projects or teams, managing technical deliverables, and mentoring, with strong problem-solving in ambiguous challenges
11
Mandatory (Education) - B.Tech or B.E. If graduated in BCA or BCS, they should have a Maters or Post graduation
12
Mandatory (Company): B2B Tech/Consulting/B2B Software
We are seeking Generative AI Developers with strong Python programming and AI/ML expertise to build, deploy, and optimize LLM-powered applications. The role involves developing RAG solutions, AI agents, and enterprise GenAI applications while collaborating with cross-functional teams.
Key Responsibilities
- Develop and enhance Generative AI applications using LLMs and AI frameworks.
- Build and optimize RAG pipelines, vector search, and AI-powered workflows.
- Design effective prompts and fine-tune models using techniques such as LoRA and QLoRA.
- Develop REST APIs and integrate AI capabilities into enterprise applications.
- Deploy, monitor, and maintain AI solutions in cloud and containerized environments.
- Ensure code quality through testing, debugging, documentation, and code reviews.
- Follow Responsible AI, security, and data governance practices.
Required Technical Skills
- Strong proficiency in Python, OOP, APIs, debugging, and software development best practices.
- Good understanding of Data Structures & Algorithms, complexity analysis, and problem-solving.
- Hands-on experience with LLMs, Prompt Engineering, RAG, AI Agents, and embeddings.
- Experience with LangChain, LangGraph, LlamaIndex, Hugging Face, or similar frameworks.
- Knowledge of vector databases, semantic/hybrid search, and retrieval architectures.
- Experience with PyTorch, TensorFlow, or Keras.
- Familiarity with Docker, Git, CI/CD, and cloud platforms (Azure/AWS/GCP).
- Understanding of AI governance, data privacy, and Responsible AI principles.
Preferred Skills
- Experience with Agentic AI frameworks (CrewAI, AutoGen, Semantic Kernel).
- Exposure to Azure AI Foundry, Databricks, or enterprise AI platforms.
- Knowledge of multimodal AI applications.
Qualifications
- Bachelor's or Master's degree in Computer Science, AI, Data Science, or a related field.
- 5 years of software development experience, including AI/ML or Generative AI projects.
- Experience building and deploying production-grade AI solutions.
Assessment Focus Areas
Candidates will be evaluated on:
- Python coding and problem-solving
- Data Structures & Algorithms
- LLMs, RAG, and Agentic AI concepts
- API development and system design
- Cloud deployment and AI solution architecture
- Senior Machine Learning Engineer – Vertex AI
- Senior Generative AI Engineer – GCP
- Google Cloud AI/ML Engineer
- MLOps Engineer – Vertex AI
- LLM Engineer – RAG, Gemini, and Vertex AI
Strong Data Scientist / AI Engineer / Generative AI Engineer profile.
2
Mandatory (Experience 1) - Must have 3+ years of hands-on experience in Data Science, Artificial Intelligence, Machine Learning, Deep Learning, NLP, or Generative AI application development.
3
Mandatory (Experience 2) - Must have strong hands-on experience in Python programming, backend development, API development, and production-grade application support.
4
Mandatory (Experience 3) - Must have experience working with Machine Learning and Deep Learning frameworks such as PyTorch, TensorFlow, Keras, or Scikit-learn.
5
Mandatory (Experience 4) - Must have hands-on experience in NLP use cases such as text classification, sentiment analysis, entity recognition (NER), semantic search, embeddings, or document understanding.
6
Mandatory (Experience 5) - Must have experience working with Large Language Models (LLMs) such as GPT, LLaMA, Mistral, Phi, Claude, Gemini, or similar models.
7
Mandatory (Experience 6) - Must have hands-on experience building or implementing Retrieval Augmented Generation (RAG) solutions, vector search, semantic search, or knowledge-based AI applications.
8
Mandatory (Experience 7) - Must have experience with Prompt Engineering and Generative AI frameworks such as LangChain, LangGraph, AI Agents, Azure OpenAI, or similar technologies.
9
Mandatory (Experience 8) - Must have experience developing, consuming, or integrating APIs using Python frameworks such as FastAPI, Flask, or similar technologies.
Mandatory (Education) - B.TECH / M.TECH from Tier 1 Colleges (IIT, NIT, BITS, IIIT, DTU, NSUT) are considered.

LLM/AI Model Engineer – Training, Evaluation & Benchmarking
Location: New Delhi, India (On-site)
Employment Type: Full-time
Experience: 3–5 years
Industry: Enterprise AI / Generative AI / Machine Learning / NLP
About QubeLabs
QubeLabs is building the next generation of Enterprise AI Systems that transform workforce operations and intelligence for the financial services industry.
Our platform combines Conversational AI, Agentic AI, workflow automation, proprietary language models and enterprise intelligence. Our flagship product, QubeLabs Workmate, serves banks, NBFCs, MFIs, wealth management firms, insurance companies and fintechs across India and Europe.
At the core of our AI platform is the Vectro Series, our proprietary family of language models designed for enterprise and financial-services use cases.
We are looking for an LLM/AI Model Engineer to build the training pipelines, datasets and evaluation infrastructure required to continuously improve the Vectro Series.
Role Overview
You will work closely with the Head of AI, Research Partners and engineering teams to implement and operationalize LLM training approaches and build a reliable model improvement lifecycle:
Data Preparation → Training → Evaluation → Benchmarking → Error Analysis → Model Improvement
The role combines hands-on LLM fine-tuning, dataset engineering, tokenization, evaluation framework development and model quality management.
Key Responsibilities
1. LLM Training & Fine-tuning
Build and maintain pipelines for training and fine-tuning open-source foundation models for the Vectro Series.
Implement supervised fine-tuning (SFT), parameter-efficient fine-tuning (PEFT), LoRA/QLoRA and domain adaptation techniques; support continued pre-training where required.
Manage training configurations, checkpoints, model versions and experiment tracking.
Optimize training workflows for computational efficiency, reproducibility and reliability.
Translate model development approaches defined with the Head of AI and Research Partners into practical, scalable training pipelines.
2. Training Data & Tokenization
Build data pipelines to collect, clean, filter, normalize, deduplicate and validate training data.
Develop instruction-tuning, SFT and domain-specific datasets for financial-services and enterprise use cases.
Implement tokenization workflows, tokenizer configuration, vocabulary management, sequence packing, truncation and padding.
Address data representation challenges involving financial terminology, numerical values and multilingual content.
Maintain dataset versioning, lineage and quality controls, incorporating identified data gaps and relevant model failures into future training data.
3. Golden Dataset & Model Evaluation
Build and maintain QubeLabs' Golden Dataset to evaluate key model capabilities and target use cases.
Develop automated evaluation pipelines and reusable evaluation harnesses, combining automated metrics with human assessment where appropriate.
Evaluate accuracy, reasoning, factuality, instruction following, safety, multilingual performance and financial-domain capabilities.
Establish consistent evaluation methodology and regression tests to measure changes across model versions.
4. Benchmarking & Error Analysis
Evaluate Vectro against relevant public benchmarks and proprietary QubeLabs BFSI/enterprise benchmarks.
Compare results with relevant open-source and commercial models using consistent evaluation conditions.
Analyze failures, identify root causes across data, training and model behavior, and translate findings into actionable improvements.
Maintain reproducible benchmark results and reports to track model strengths, limitations and progress.
5. Model Quality & Release
Define model release quality gates and ensure each major release meets agreed evaluation and benchmark criteria.
Support red-teaming, robustness and safety testing.
Coordinate with the Head of AI, Research Partners and engineering teams to ensure validated model improvements are suitable for integration into production systems.
Experience
3–5 years in Machine Learning, Deep Learning, NLP, Generative AI or related fields.
Hands-on experience training or fine-tuning LLMs using open-source foundation models.
Practical experience building training pipelines, preparing datasets and evaluating model performance.
Experience with GPU-based training; distributed training is an advantage.
Financial-services, multilingual AI or enterprise AI experience is preferred.
Required Skills
LLM & Machine Learning: Large Language Models, Transformer architectures, Generative AI, NLP, Deep Learning, SFT, PEFT, LoRA/QLoRA, fine-tuning and continued pre-training.
Frameworks & Infrastructure: Python, PyTorch, Hugging Face Transformers, Hugging Face Tokenizers, GPU computing, experiment tracking and ML pipelines.
Data & Tokenization: Tokenization, tokenizer configuration, text preprocessing, sequence packing, dataset construction, data quality, versioning and lineage.
Evaluation & Benchmarking: Golden datasets, LLM evaluation, evaluation harnesses, automated and human evaluation, LLM-as-a-Judge, public and domain-specific benchmarks, regression testing and error analysis.
The role requires the ability to build both the model training pipeline and the measurement system that determines whether the model has actually improved.
What Success Looks Like
Reliable and reproducible training and fine-tuning pipelines for the Vectro Series.
High-quality training datasets, tokenization workflows and a comprehensive Golden Dataset.
Automated evaluation and benchmarking infrastructure with measurable model quality gates.
Clear, reproducible evidence of model performance against public, proprietary BFSI and relevant competing-model benchmarks.
A continuous improvement loop that converts evaluation findings into measurable gains across successive Vectro releases.
Key Performance Indicators
Training pipeline reliability and experiment throughput.
Training data quality and coverage.
Golden Dataset and evaluation coverage.
Benchmark reproducibility and model performance improvement.
Accuracy of error diagnosis and effectiveness of regression detection.
Time required to complete training, evaluation and benchmarking cycles.
Compliance with model release quality criteria.
Why Join QubeLabs?
Build the proprietary Vectro Series of enterprise and financial-services language models.
Develop model training, evaluation and benchmarking infrastructure from the ground up.
Work closely with the Head of AI and Research Partners.
Solve real-world AI challenges across financial services.
Develop proprietary datasets, benchmarks and model improvement systems.
Contribute directly to production AI systems serving customers across India and Europe.
Preferred Candidate Profile
B.Tech, M.Tech, MS or PhD in Computer Science, AI, ML, Mathematics or a related field.
3–5 years of relevant experience with strong practical skills in Python, PyTorch, LLM fine-tuning and training data pipelines.
Working knowledge of tokenization, evaluation datasets, benchmarking and regression testing.
Experience with open-source models such as Llama, Qwen, Mistral or similar is preferred.
Ability to independently build reliable model training and evaluation systems from scratch.
QubeLabs is an equal opportunity employer and values diversity and inclusion. We encourage applications from qualified candidates regardless of background or identity.







