[Remote] Staff Machine Learning Engineer

Note: The job is a remote job and is open to candidates in USA. webAI is the first end-to-end private AI platform. They are seeking an experienced Staff Machine Learning Engineer with a strong background in Large Language Models and Mixture of Experts to lead core product initiatives and optimize AI models.

Responsibilities

• Lead the development and optimization of Large Language Models and Mixture of Experts models
• Collaborate with cross-functional teams to integrate ML models into our platform
• Conduct cutting-edge research in machine learning, with a focus on improving the performance and efficiency of LLMs
• Stay abreast of the latest advancements in AI and ML, and apply this knowledge to improve our models and methodologies
• Mentor junior engineers and contribute to the team’s knowledge sharing and best practices

Skills

• Advanced degree (Ph.D. preferred) in Computer Science, or a related field
• Proven track record of building and innovations through publications or industry experience
• Minimum of 6 years of experience in machine learning, with specific expertise in Large Language Models and Mixture of Experts
• Strong programming skills in Python and machine learning frameworks like TensorFlow and/or PyTorch
• Demonstrated ability to lead complex projects and work collaboratively in a team environment
• Excellent problem-solving skills and a passion for innovation
• Experience with cloud computing services (AWS, Azure, GCP)
• Knowledge of Big Data technologies (Hadoop, Spark)
• Familiarity with containerization and orchestration technologies (Docker, Kubernetes)
• Publications or presentations in recognized Machine Learning journals or conferences

Benefits

• Comprehensive health, dental, and vision benefits package
• 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

Company Overview

The leader in private AI. It was founded in 2020, and is headquartered in Austin, Texas, USA, with a workforce of 51-200 employees. Its website is https://www.webai.com.

Company H1B Sponsorship

webAI has a track record of offering H1B sponsorships, with 2 in 2026, 9 in 2025, 2 in 2024. Please note that this does not guarantee sponsorship for this specific role.