Staff Machine Learning Engineer
webAISovereign AI company
Austin, United StatesLead
Oxcart Ventures
Atreides Management
Data & AI
About the role
TL;DR
Lead ML initiatives for LLMs and MoEs, focusing on optimization and deployment.
- •webAI is seeking an experienced Staff Machine Learning Engineer with a strong background in Large Language Models (LLMs) and/or Mixture of Experts (MoEs).
- •You will lead core product initiatives across on-device inference optimization, quantization, RAG, agentic framework and/or tool calling.
- •Key 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.
- •Requirements Advanced degree (Ph.D. preferred) in Computer Science, or a related field.
- •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.
Required skills
PythonLLMsTensorFlowPyTorchMentoring
Nice-to-have skills
AWSAzureGoogle CloudSparkDockerKubernetes
Domain expertise
aideveloper-tools
Benefits & perks
Competitive salary, 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
Tech stack
PythonTensorFlowPyTorchAWSAzureGoogle CloudDockerKubernetesSparkLLMs
