4 months ago
Responsibilities
- Lead the development and optimization of Large Language Models and Mixture of Experts models.
- Lead core product initiatives across on-device inference optimization, quantization, RAG, agentic frameworks, and tool calling.
- Integrate machine learning models into the platform in collaboration with cross-functional engineering teams.
- Conduct machine learning research focused on improving the performance and efficiency of LLMs.
- Lead the sub-team and manage end-to-end delivery of initiatives.
- Mentor junior engineers and contribute to team knowledge sharing and best practices.
- Monitor advances in AI and machine learning and apply relevant developments to models and methodologies.
Requirements
- Advanced degree, preferably a Ph.D., in Computer Science or a related field.
- At least 6 years of machine learning experience, including specific expertise in Large Language Models and Mixture of Experts.
- Proven track record of building and innovating through publications or industry experience.
- Strong programming skills in Python and experience with TensorFlow and/or PyTorch.
- Ability to lead complex projects and work collaboratively across teams.
- Strong problem-solving skills and a passion for innovation.
Benefits
- Competitive salary and performance-based incentives.
- Comprehensive health, dental, and vision benefits, including a high-deductible HSA option with company contribution.
- 401(k) match for U.S.-based employees.
- Flexible PTO policy.
- Parental leave for eligible employees.
- Continuing education support.
- Basic company-paid life insurance and supplemental life insurance.
- Company-paid short- and long-term disability coverage.
- Free parking for in-office employees.
- Benefits may vary for employees hired outside the United States based on local law, country-specific requirements, and the employment platform or entity.
Tech Stack
Categories
About webAI
webAI builds an end-to-end platform for training, deploying, and operating private AI models on infrastructure customers control, including edge devices. It sells enterprise software and developer SDKs/APIs used by public-sector, defense, aviation, and manufacturing teams to run specialized models with data sovereignty. Founded in 2020 and headquartered in Austin, Texas, the privately held company focuses on distributed AI/LLM infrastructure across local and cloud environments.