Staff Machine Learning Engineer
webAI · Austin, TX · Remote Office
Applied AIremotestaff
At a glance
Highlights
- Lead core LLM and Mixture of Experts initiatives for end-to-end private AI platform.
- Drive on-device inference optimization, RAG, and agentic tool calling.
- Bring specialized, sovereign AI to enterprise and government data.
What they offer
- Competitive salary and performance-based incentives
- Comprehensive health, dental and vision benefits including HSA option
- 401k match (US-based only)
- Flexible PTO policy
- Parental leave for eligible employees
- Continuing education support
- Company-paid life insurance plus supplemental life insurance
- Company-paid short and long term disability
- Free parking for in-office employees
Ideal candidate
An experienced ML innovator passionate about efficient, specialized AI who thrives leading complex initiatives, mentoring others, and turning cutting-edge research into real-world impact.
Requirements
- Advanced degree (Ph.D. preferred) in Computer Science or related field
- Minimum of 6 years of experience in machine learning with expertise in LLMs and Mixture of Experts
- Proven track record of building innovations through publications or industry experience
- Strong programming skills in Python and ML frameworks like TensorFlow and/or PyTorch
- Demonstrated ability to lead complex projects and work collaboratively
- Excellent problem-solving skills and passion for innovation
Lab profile
About webAI
Building distributed, private AI infrastructure for personalized, edge-deployed enterprise and government AI.
What they build
- Distributed AI infrastructure
- End-to-end private AI platform
Known for
- Distributed edge processing
- Private and sovereign AI
- Specialized enterprise intelligence
- Efficient and sustainable models
Posted · Last seen