Software Engineer 2, Machine Learning

This is a fully remote position, open to applicants in India.

📋 Description

• Collaborate with product managers and business stakeholders to gain insight into business challenges and convert them into machine learning problems along with actionable project requirements.

• Research and devise solutions utilizing statistical modeling, machine learning, and generative AI/LLMs.

• Partner with machine learning engineers to develop and enhance solutions.

• Collaborate with Applied AI/ML engineers to create LLM-powered agentic systems for real-world product applications.

• Design, develop, and optimize agent pipelines and APIs.

• Create tests for agent behaviors and contribute to evaluation frameworks.

• Deploy machine learning solutions into scalable production environments on cloud platforms such as AWS, Azure, and GCP.

• Ensure that deployed solutions are effective.

• Remain updated on advancements in AI/ML, publications, best practices, evaluation methodologies, technology stacks, and relevant tools.

⛳️ Requirements

• A minimum of 2 years of full-time experience as an ML engineer in tier-1 product-based companies.

• Adopting an AI-first approach for addressing challenges (MIMP).

• Practical experience in building statistical and machine learning models, demonstrating expertise in both traditional methods (regression, classification, clustering, tree-based models) and deep learning techniques (neural networks, CNNs, RNNs).

• Strong skills in prompt engineering.

• Proficient debugging and online research abilities.

• Effective communication skills.

• Strong problem-solving skills, including the capability to tackle challenging data structures and algorithms problems.

• Practical experience in building and deploying AI Agents using LangChain, LangGraph, OpenAI API, Gemini API, Anthropic API, etc. (preferred).

• Experience with multi-step agent workflows, orchestration, and evaluating agent reliability (preferred).

• Familiarity with vector databases (Pinecone, Weaviate) and semantic search (preferred).

• Experience in fine-tuning open-weight language models and deploying and maintaining them in scalable production environments on cloud platforms like AWS, Azure, or GCP (preferred).

🏝️ Benefits

• Permanent remote / work-from-home culture

• Opportunity to work with a cutting-edge AI technology stack

• An AI-first approach to tackling real-world problems

• Complete ownership and trust from leadership

• Your ideas and concerns are actively considered by the leadership team

• Freedom to experiment, fail, and learn

• Rapid career advancement for high performers

• Collaborate with a high-caliber team of engineers and industry experts

• Gain insights into how a fast-growing startup builds and scales