Staff Machine Learning Engineer

WebAI
Austin
Workplace: OnsiteFull timeFunction: Data Science & Machine LearningExperience: 6+ yearsEducation: phdSkills: ["Problem-solving","Innovation","Collaboration","Mentorship","Ownership","Tenacity","Humility","Transparency"]

Lead core product initiatives in private AI, focusing on Large Language Models and Mixture of Experts. Drive on-device inference optimization, quantization, RAG, agentic frameworks, and tool calling, while integrating ML models into the platform. Conduct ML research to improve LLM performance and efficiency, and stay current with AI advancements. Collaborate cross-functionally and mentor junior engineers, contributing to best practices.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
WebAI
WebAI
1 month ago

Staff Machine Learning Engineer

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 11 hours agoStatus: Live

Job Summary

Lead core product initiatives in private AI, focusing on Large Language Models and Mixture of Experts. Drive on-device inference optimization, quantization, RAG, agentic frameworks, and tool calling, while integrating ML models into the platform. Conduct ML research to improve LLM performance and efficiency, and stay current with AI advancements. Collaborate cross-functionally and mentor junior engineers, contributing to best practices.
Location: Austin
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Lead development and optimization of Large Language Models and Mixture of Experts models.
  • •Collaborate with cross-functional teams to integrate ML models into the platform.
  • •Conduct ML research to improve performance and efficiency of LLMs.
  • •Apply latest AI/ML advancements to improve models and methodologies.
  • •Mentor junior engineers and contribute to team knowledge sharing and best practices.

Pay and Benefits

Perks:Health InsuranceDentalVision401kEquityWellness StipendLearning BudgetParkingFlexible TimeParental LeaveLife Insurance

Key Requirements

  • •Ph.D. (preferred) in Computer Science or a related field.
  • •6+ years of machine learning experience with expertise in Large Language Models and Mixture of Experts.
  • •Strong programming skills in Python and ML frameworks such as TensorFlow and/or PyTorch.
  • •Proven track record of building and innovations through publications or industry experience.
  • •Ability to lead complex projects and collaborate effectively in a team environment.
Experience:6+ yearsMachine learningLarge language modelsMixture of expertsLLM researchRAG
Education:PhD / Doctorate in Computer Science
Skills:Problem-solvingInnovationCollaborationMentorshipOwnershipTenacityHumilityTransparency
Tech Stack:PythonTensorFlowPyTorchAWSAzureGCPHadoopSparkDockerKubernetes

Company Brief

WebAI
Provides an AI platform to build, deploy, and manage web-based AI applications, offering tools for natural language, vision, and web automation so businesses and developers can integrate intelligent features into websites and products.
Industry: AI & Machine Learning
Website