Senior Machine Learning Engineer
Location: Onsite - Bengaluru
Company: Pibit.ai : Y Combinator backed Insurtech Startup
About Pibit.ai
Pibit.ai is transforming the underwriting landscape with Generative AI. Our SaaS solutions help insurance companies in the US make smarter, faster decisions by optimizing underwriting processes, reducing risk, and improving premiums. As we expand, we’re looking for a Senior Machine Learning Engineer to build NLP, CV & LLM based solutions and generate value for our customers.
Position Overview
As a Senior Machine Learning Engineer, you will design, implement, and optimize AI solutions while ensuring model success. You will lead the ML lifecycle from development to deployment, collaborate with cross-functional teams, and enhanceAI capabilities to drive innovation and impact.
Key Responsibilities:
● Design and implement AI product features.
● Maintain and optimize existing AI systems.
● Train, evaluate, deploy, and monitor ML models.
● Design ML pipelines for experiment, model, and feature management.
● Implement A/B testing and scalable model inferencing APIs.
● Optimize GPU architectures, parallel training, and fine-tune models for improved performance.
● Deploy LLM solutions tailored to specific use cases.
● Ensure DevOps and LLMOps best practices using Kubernetes, Docker, and orchestration frameworks.
Technical Requirements:
● LLM & ML: Hugging Face OSS LLMs, GPT, Gemini, Claude, Mixtral, Llama
● LLMOps: MLFlow, Langchain, Langgraph, LangFlow, Langfuse, LlamaIndex, SageMaker, AWS Bedrock, Azure AI
● Databases: MongoDB, PostgreSQL, Pinecone, ChromDB
● Cloud: AWS, Azure
● DevOps: Kubernetes, Docker
● Languages: Python, SQL, JavaScript
● Certifications (Bonus): AWS Professional Solution Architect, AWS Machine
Learning Specialty, Azure Solutions Architect Expert
What You'll Do:
● Collaborate with cross-functional teams to design and build scalable MLsolutions.
● Implement state-of-the-art ML techniques, including NLP, Generative AI, RAG, and Transformer architectures.
● Deploy and monitor ML models for high performance and reliability.
● Innovate through research, staying ahead of industry trends.
● Build scalable data pipelines following best practices.
● Present key insights and drive decision-making.
What You Need to Succeed:
● Master’s degree or equivalent experience in Machine Learning.
● 5+ years of industry experience in ML, software engineering, and data engineering.
● Proficiency in Python, PyTorch, TensorFlow, and Scikit-learn.
● Strong programming skills in Python and JavaScript.
● Hands-on experience with ML Ops practices.
● Ability to work with research and product teams.
● Excellent problem-solving skills and a track record of innovation.
● Passion for learning and applying the latest technological advancements.
Why Join Us?
● Work directly with experienced founders.
● Be part of a high-energy team that works hard and celebrates success.
● Shape the future of AI-driven automation in the insurance industry.
Website - https://www.pibit.ai/
Skills Required
- Master's degree or equivalent experience in Machine Learning
- 5+ years industry experience in ML, software engineering, and data engineering
- Proficiency in Python
- Proficiency in PyTorch
- Proficiency in TensorFlow
- Proficiency in Scikit-learn
- Strong programming skills in JavaScript
- Experience with LLMs (Hugging Face OSS LLMs, GPT, Gemini, Claude, Mixtral, Llama)
- Experience with LLMOps tools (MLFlow, Langchain, Langgraph, LangFlow, Langfuse, LlamaIndex)
- Experience with cloud ML platforms (SageMaker, AWS Bedrock, Azure AI)
- Experience with databases (MongoDB, PostgreSQL, Pinecone, ChromDB)
- Experience with AWS and Azure cloud
- Experience with Kubernetes and Docker
- Experience optimizing GPU architectures, parallel training, and fine-tuning models
- Hands-on experience with MLOps practices
- AWS Professional Solutions Architect, AWS Machine Learning Specialty, Azure Solutions Architect Expert
What We Do
NextHire Consulting is an AI-driven recruiting platform that streamlines the hiring process for companies. By leveraging AI agents for sourcing, screening, and interviewing, the platform enables teams to focus on pre-qualified finalists. It provides data-driven insights into candidate soft skills and behavioral styles, aiming to disrupt traditional recruitment models with efficient, automated, and science-based talent acquisition solutions for businesses of all sizes.
