Senior Machine Learning Engineer

About the Company

Pibit.ai is a Y Combinator backed Insurtech startup 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.

Responsibilities

• Design and implement AI product features and maintain 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
• Deploy LLM solutions tailored to specific use cases
• Ensure DevOps and LLMOps best practices using Kubernetes and Docker
• Collaborate with cross-functional teams to build scalable ML solutions
• Implement state-of-the-art techniques including NLP, Generative AI, RAG, and Transformer architectures

Requirements

• 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
• Experience with LLMs (GPT, Gemini, Claude, Llama) and LLMOps tools (Langchain, LlamaIndex, etc.)
• Experience with databases like MongoDB, PostgreSQL, Pinecone, or ChromDB
• Proficiency with cloud platforms (AWS, Azure) and DevOps tools (Kubernetes, Docker)

Preferred Qualifications

• AWS Professional Solution Architect certification
• AWS Machine Learning Specialty certification
• Azure Solutions Architect Expert certification