Staff Machine Learning Engineer
webAI · Austin, TX · Remote Office
CHECKED 18H AGO
Applied AIremotestaff
At a glance
Highlights
- Lead core initiatives across on-device inference, RAG and agentic frameworks
- Build specialized private AI that learns directly from enterprise data
- Drive efficient AI challenging the assumption that big data is required
What they offer
- Competitive salary
- Comprehensive health, dental, and vision benefits
- 401(k) match
- Equity options
- $200/month Health & Wellness stipend
- Continuing Education support
- $500/year Function Health subscription
- Free parking for in-office employees
- Flexible Time Off (FTO)
- Parental leave for eligible employees
- Supplemental life insurance
Ideal candidate
Seasoned ML leader passionate about efficient, sovereign AI who thrives owning complex projects end-to-end, mentoring others, and turning cutting-edge research into real-world products.
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 MoEs
- Strong programming skills in Python and TensorFlow and/or PyTorch
- Proven track record of innovations through publications or industry experience
- Demonstrated ability to lead complex projects and collaborate in a team environment
- 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