AI/ML Engineer for LLM Development

Build, deploy, and optimize large language model based systems that power high-quality AI training and evaluation. As a core member of the AI/ML engineering team, you will design scalable APIs and services, integrate advanced model orchestration frameworks, and apply quantitative expertise to improve model performance and reliability. You will work remotely with cross-functional engineers and domain experts at micro1, an AI data lab focused on turning real-world subject matter expertise into training data, evaluations, and feedback loops for frontier models.

• Design, implement, and deploy AI/ML solutions centered on large language models using Python and AWS infrastructure.
• Architect and maintain scalable APIs and microservices, using FastAPI to ensure high performance and reliability.
• Integrate LangChain and other advanced AI frameworks to enable multi-step model interactions and complex reasoning workflows.
• Translate mathematical, statistical, or physics-based problem statements into concrete engineering designs and implementable systems.
• Apply core math, statistics, or physics knowledge to improve model interpretability, robustness, and overall performance.
• Maintain strong code quality, documentation, and participate in peer reviews within a fast-paced core team environment.
• Communicate technical designs and project status clearly in written reports and verbal discussions.

Required

• Advanced degree or demonstrable expertise in mathematics, quantitative sciences, statistics, or physics.
• Expert proficiency in Python for AI/ML engineering.
• Hands-on experience deploying, fine-tuning, and integrating large language models.
• Practical experience with AWS services and cloud native machine learning workflows.
• Proven use of LangChain for chaining LLMs and orchestrating multi-step reasoning.
• Solid understanding of FastAPI for building scalable APIs.
• Excellent written and verbal communication skills, able to explain complex technical topics clearly.

Preferred

• Experience delivering AI/ML solutions in production 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.

• Employment type: Full-time.
• Location: Remote.
• Role is part of the core engineering team and operates in a fast paced, remote-first environment.

• Listed compensation range: $400, 000 to $800, 000 per year.
• National pay range for base salary: $200, 000 to $260, 000 USD.
• All employees are eligible for equity compensation. Employees may also receive performance based bonuses, subject to role and company policies.
• Benefits include up to 100% reimbursement for health insurance premiums, paid time off, a 401(k) plan with company match, and other benefits designed for a remote first workforce.

• All qualified applicants will receive consideration without regard to race, color, religion, sex including pregnancy, sexual orientation, or gender identity, national origin, age, disability, genetic information, veteran status, or other characteristics protected by applicable law.
• If you need assistance or a reasonable accommodation during the application process, contact support@micro1.ai for help.