Job Description

Staff Machine Learning Engineer - LLMs / Mixture-of-Experts

(Hybrid Austin, US Citizens Only)

The Role

Are you excited by the challenge of pushing the boundaries of what modern AI models can do - especially when data is limited? A fast-growing AI platform is looking for a Staff Machine Learning Engineer to help shape the next generation of large-scale intelligent systems.

In this role, you’ll take the lead on developing advanced Large Language Models (LLMs) and Mixture-of-Experts (MoE) architectures, driving innovation that directly influences product capabilities and performance. If you thrive at the intersection of research and real-world impact, you’ll feel right at home here.

This hybrid role is based in Austin, Texas , with relocation support available for qualified candidates. U.S. citizenship is required due to work involving security-sensitive projects .

What You’ll Work On

• Architect, train, and optimize cutting-edge LLMs and MoE-based systems.
• Experiment with novel algorithms to improve efficiency, scalability, and model performance.
• Collaborate closely with engineering and product teams to deploy ML capabilities into production.
• Contribute to pioneering research in ML and NLP, driving methodological advancements.
• Mentor engineers and help shape technical best practices across the organisation.

What We’re Looking For

• Advanced degree in Computer Science or a related field (PhD preferred).
• 6+ years of industry experience building and deploying machine learning models at scale.
• Deep expertise in LLMs, Mixture-of-Experts architectures , and modern ML frameworks such as PyTorch or TensorFlow.
• Demonstrated innovation through impactful research, patents, or production-grade ML systems.
• Ability to lead complex, cross-functional technical initiatives.
• Strong problem-solving skills and a passion for pushing the boundaries of AI.

Bonus Skills

• Publications or conference presentations at leading ML/NLP venues such as NeurIPS, ICML, ICLR, AAAI, EMNLP, NACL, ACL, EACL, CoNLL , or similar.
• Experience with cloud platforms (AWS, Azure, GCP) and distributed computing tools (Spark, Hadoop).
• Familiarity with containerization and orchestration (Docker, Kubernetes).

Why You’ll Love It

You’ll join a team that embodies transparency, ownership, tenacity, and humility - values that guide both technical decision-making and collaboration.

You’ll also enjoy:

• Competitive compensation and performance incentives
• Comprehensive medical, dental, and vision benefits
• Monthly wellness stipend + annual continuing education credit
• A flexible work environment and unlimited approved PTO
• Parental and bereavement leave and other employee support programs

This role is hybrid in Austin with relocation assistance offered for the right candidate.

Only U.S. citizens are eligible due to the security clearance requirements.

Job Tags

Similar Jobs