Staff Machine Learning Engineer

WebAI

Remote regions

US

Benefits

Unlimited PTO

Similar Jobs

See all

Role:

  • 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.

Qualifications:

  • 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 LLMs and MoEs.
  • Strong programming skills in Python and machine learning frameworks like TensorFlow and/or PyTorch.

Benefits:

  • Competitive salary with comprehensive health, dental, and vision benefits package.
  • 401(k) match, equity options, and flexible time off.
  • Parental leave for eligible employees and supplemental life insurance.

WebAI

webAI is an end-to-end private AI platform that enables enterprises and governments to bring AI to their data, powering specialized intelligence trained on their own knowledge. The company is a dynamic, fast-growing team fostering an exciting and growth-oriented work culture, committed to truth, ownership, tenacity, and humility.

Apply for This Position