Apply on Employer Site

webAI · 5 hours ago

Staff Machine Learning Engineer

Austin, Texas Metropolitan Area

Full-time

Hybrid

Senior Level, Lead/Staff

6+ years exp

webAI is a software company that is building a decentralized AI development platform. They are seeking an experienced Staff Machine Learning Engineer with a strong background in Large Language Models and Mixture of Experts to lead the development and optimization of advanced AI models.

Computer Software

H1B Sponsor Likely

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

Qualification

Large Language ModelsMixture of ExpertsMachine LearningPythonTensorFlowPyTorchCloud ComputingInnovationProblem-solvingTeam Collaboration

Required

Advanced degree (Ph.D. preferred) in Computer Science, or a related field

Proven track record of 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

Preferred

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

Competitive salary and performance-based incentives.

Comprehensive health, dental, and vision benefits package.

$200/mos Health and Wellness Stipend

$400/year Continuing Education Credit

Flexible work week

Free parking, for in-office employees

Unlimited Approved PTO

Parental, Bereavement Leave

Supplemental Life Insurance

Company

webAI

webAI is designed to streamline the training, deployment, and execution of AI models by offering a unified execution layer for AI that seamlessly integrates cloud-based services and local devices.

Founded in 2020

Austin, Texas, US

51-200 employees

https://linktr.ee/thewebai

H1B Sponsorship

webAI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (9)

2024 (2)