Machine Learning Engineer 2
We are looking for a Machine Learning Engineer 2 (Onsite, Bengaluru) to design, build, and scale AI solutions that power our underwriting platform for commercial insurance carriers and MGAs in the U.S. market. You will work on NLP, CV, and LLM use cases that directly impact how insurers process submissions and make decisions. This is a hands on role for someone who enjoys taking models from experimentation to stable, production grade systems. You will own parts of the ML lifecycle, partner closely with product and engineering, and help advance Pibit.ai's AI capabilities.
Key Responsibilities
- Design and implement AI product features that support core underwriting workflows.
- Maintain, optimize, and improve existing AI systems for performance, quality, and cost.
- Train, evaluate, deploy, and monitor machine learning and LLM models in production.
- Build ML pipelines for experimentation, model management, and feature management.
- Implement A/B testing frameworks and scalable inference APIs for online and batch use cases.
- Optimize GPU usage, parallel training strategies, and fine tuning approaches to improve model efficiency.
- Deploy LLM based solutions tailored to specific document, workflow, and retrieval use cases.
- Apply DevOps and LLMOps best practices using Kubernetes, Docker, and supporting orchestration frameworks.
- Collaborate with cross functional teams to define problems, translate them into ML solutions, and present insights that guide decisions.
- Stay current with advances in NLP, Generative AI, RAG, and Transformer architectures and bring relevant techniques into the product.
Requirements
- Master's degree or equivalent experience in Machine Learning, Computer Science, or a related field.
- 3+ years of industry experience across machine learning, software engineering, and data engineering.
- Strong programming skills in Python, with experience in PyTorch, TensorFlow, and Scikit learn.
- Hands on experience building and operating ML systems in production, including MLOps practices.
- Experience with LLMs and ML stacks such as Hugging Face models, GPT, Gemini, Claude, Mixtral, or Llama.
- Familiarity with LLMOps tools and frameworks such as MLflow, LangChain, LangGraph, LangFlow, Langfuse, LlamaIndex, SageMaker, AWS Bedrock, or Azure AI.
- Experience with databases and vector stores such as MongoDB, PostgreSQL, Pinecone, and ChromaDB.
- Proficiency with cloud platforms such as AWS or Azure and container technologies such as Kubernetes and Docker.
- Working knowledge of SQL and JavaScript for integrations and end to end solution development.
- Strong problem solving skills, ability to work with research and product teams, and a track record of innovation.
- Onsite role in Bengaluru. Remote work is not available for this position.
Why Join Us
- Work directly with experienced, founder led leadership on meaningful AI initiatives.
- Contribute to an AI first underwriting platform used by leading U.S. insurance companies.
- Join a high energy team that values ownership, learning, and fast iteration.
- Shape the future of AI powered automation in the insurance industry with real world impact.
- Grow your career in an environment that encourages experimentation, technical depth, and visible contributions.
Apply today to take the next step in your journey!
Ready to optimize
Loss ratios, account win rate, and throughput?
611 Gateway Blvd Suite 120
South San Francisco, CA, 94080,
United States
