WebAI
Staff Machine Learning Engineer
Austin, TX +1 · Staff · Full-time · Posted Jul 10, 2026
- Location
- Austin, TX
Remote Office - Experience
- Minimum of 6 years in machine learning, with specific expertise in Large Language Models and Mixture of Experts.
- Education
- Advanced degree in Computer Science or a related field; Ph.D. preferred.
About the job
Leads development, optimization, and integration of LLM and Mixture of Experts models, including work on inference efficiency and related product initiatives.
What you'll do
- Lead development and optimization of LLM and Mixture of Experts models, focusing on performance and efficiency.
- Collaborate across engineering teams to integrate models into the platform and deliver core product initiatives.
- Mentor engineers and share best practices.
What they ask for
- At least 6 years of machine learning experience, with specific expertise in LLMs and Mixture of Experts.
- Strong Python skills and experience with TensorFlow and/or PyTorch.
- Track record of building and innovation through publications or industry experience.
- Ability to lead complex projects and collaborate in a team.
Nice to have
- Ph.D. in Computer Science or a related field.
- Experience with cloud computing, Big Data technologies, or containerization and orchestration.
Tools
Python, TensorFlow, PyTorch, AWS, Azure, GCP, Hadoop, Spark
AI focus
Develop and optimize LLMs and Mixture of Experts; improve model performance and efficiency; lead initiatives involving on-device inference optimization, quantization, RAG, agentic frameworks, and tool calling.
Model-written summary · the original posting is authoritative