Software Engineer 2 , Machine Learning

What will excite us?

• MIMP: An AI-first approach for solving problems.
• Hands-on experience building statistical and machine learning models, with proven expertise across traditional methods (regression, classification, clustering, tree-based models) and deep learning techniques ( neural networks, CNNs, RNNs).
• Solid prompt engineering skills.
• Hands-on experience building and deploying AI Agents using LangChain, LangGraph, OpenAI API, Gemini API, Anthropic API, etc.
• Hands-on experience with multi-step agent workflows, orchestration, and evaluating agent reliability.
• Hands-on experience with vector databases (Pinecone, Weaviate) and semantic search.
• Hands-on experience fine-tuning open-weight language models and successfully deploying and maintaining them in scalable production environments on cloud platforms (e.g., AWS, Azure, GCP).
• Good debugging and online searching skills.
• Good communication skills.
• Good problem-solving skills (ability to solve hard-level data structures and algorithms problems).

What will you be doing?

• Collaborate closely with product managers and business stakeholders to gain a deep understanding of business challenges, translating them into well-defined machine learning problems and actionable project requirements that drive business growth and enhance learner experiences.
• Leverage your expertise in statistical modeling, machine learning, and generative AI/LLMs to research and design optimal solutions for the identified machine learning problems.
• Collaborate closely with machine learning engineers to develop and refine optimal solutions for the identified machine learning problems.
• Closely work with Applied AI/ML engineers building LLM-powered agentic systems for real product use cases.
• Design, develop, and optimize agent pipelines/APIs, build tests for agent behaviors, and contribute to evaluation frameworks.
• Take ownership of deploying the developed machine learning solutions into scalable production environments (on cloud platforms like AWS, Azure, GCP) and ensure these deployed solutions are effective.
• Staying informed about recent developments in AI/ML, including key publications, best practices, evaluation methodologies, technology stacks, and relevant tools.

Why Will You Enjoy Working with Us?

• Permanent remote / work-from-home culture
• Opportunity to work on a cutting-edge AI technology stack
• An AI-first approach to solving real-world problems
• Complete ownership and trust from leadership
• Your ideas and concerns are actively heard by the leadership team
• Freedom to experiment, fail, and learn
• Rapid career progression for high performers
• Work with a high-caliber team of engineers and industry experts
• Learn how a fast-growing startup builds and scales