Member of Technical Staff, AI/ML Engineering

Micro1; India; Remote; INR 18,779,000 - 24,413,000; Full time

Benefits offered by this job

Equity compensation

Performance-based bonuses

Health-insurance premium reimbursement

Paid time off

401(K) with company match

Remote-first workforce

Job summary

micro1 seeks a Member of Technical Staff, AI/ML Engineering, to lead AI system development and optimization. The role centers on designing, implementing, and refining LLM-based solutions with Python, AWS, and LangChain.

You’ll build scalable APIs with FastAPI, collaborate across teams to translate mathematical concepts into engineering outcomes, and uphold strong code quality with clear documentation in a remote-first environment.

Qualifications

• Advanced degree in math, quantitative sciences, statistics, or physics.
• Proficiency in Python for AI/ML engineering and LLM deployment.
• Hands-on experience with AWS and building scalable AI systems.
• Experience with LangChain for multi-step reasoning and chaining LLMs.
• Strong written and verbal communication skills.

Responsibilities

• Design, build, and deploy AI/ML solutions focusing on LLMs using Python and AWS.
• Architect scalable APIs and microservices with FastAPI.
• Integrate LangChain and advanced AI frameworks for model interaction.
• Translate complex mathematical or statistical problems into engineering solutions.
• Apply math, statistics, or physics to improve model performance and reliability.
• Maintain code quality, documentation, and peer reviews in a fast-paced team.
• Communicate progress clearly through reports and discussions.

Skills

Python for AI/ML engineering

LangChain

LLMs deployment

AWS

API design

Clear communication

Education

Advanced degree in math/quantitative sciences or physics

Tools

FastAPI

LangChain

AWS

Job description

Job Type: Full-time

Location: Remote

Job Summary

Join our team as a Member of Technical Staff, AI/ML Engineering, and drive the development of cutting-edge AI systems. Work alongside talented engineers to design, implement, and optimize advanced LLM-based solutions, leveraging your expertise in quantitative fields to create impactful, scalable products.

Key Responsibilities

• Design, build, and deploy AI/ML solutions with a focus on large language models (LLMs) using Python and AWS infrastructure.
• Architect scalable APIs and microservices utilizing FastAPI, ensuring robust, high-performance systems.
• Integrate LangChain and advanced AI frameworks to deliver innovative model interaction capabilities.
• Collaborate with cross-functional teams to translate complex mathematical, statistical, or physics-based problem statements into actionable engineering solutions.
• Apply core math, statistics, or physics knowledge to improve model performance, interpretability, and reliability.
• Maintain high standards of code quality, documentation, and peer reviews within a fast-paced core team.
• Communicate technical concepts and project progress clearly through both written reports and verbal discussions.

Required Skills and Qualifications

• Advanced degree or demonstrable expertise in math, quantitative sciences, statistics, or physics.
• Expert proficiency in Python for AI/ML engineering.
• Hands-on experience deploying, fine-tuning, and integrating LLMs.
• Strong background in AWS services and cloud-native ML workflows.
• Proven track record using LangChain for chaining LLMs and orchestrating multi-step reasoning.
• Solid understanding of FastAPI for building scalable APIs.
• Exceptional written and verbal communication skills; ability to articulate complex technical topics with clarity.

Preferred Qualifications

• Experience delivering AI/ML solutions in production environments at scale.
• Publication record or open-source contributions in AI/ML, applied mathematics, or computational science.
• Familiarity with advanced prompt engineering or AI safety best practices.

Compensation & Benefits Notice

The national pay range for this full-time position is base salary of $200,000 -$260,000 USD. All employees are eligible for equity compensation, and employees may also receive performance-based bonuses, dependent on role and subject to company policies.

micro1 provides a comprehensive benefits package, including up to 100% reimbursement for health-insurance premiums, paid time off, a 401(K) plan with a company match, and additional benefits designed to support a high-performing, remote-first workforce.

• up to 100% reimbursement for health-insurance premiums
• paid time off
• a 401(K) plan with a company match
• additional benefits designed to support a high-performing, remote-first workforce

micro1 is an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex (including pregnancy, sexual orientation, or gender identity), national origin, age, disability, genetic information, veteran status, or any other characteristic protected by applicable local laws, regulations and ordinances.

If you need assistance and/or a reasonable accommodation during the application process, reach out to [e-mail].

Our hiring process utilizes artificial intelligence tools to assist in candidate screening and assessment. Our AI tools are designed to complement, not replace, human decision-making.

Disclaimer

The information contained in this job posting, including but not limited to role responsibilities, qualifications, compensation, and benefits, is provided for informational purposes only and does not constitute a binding offer of employment. micro1 reserves the right to amend, modify, or withdraw any portion of this posting at its sole discretion and without prior notice. All employment decisions are made in accordance with applicable laws and regulations.