Skip to main content
Back to jobs

Staff Machine Learning Engineer

External
webai logoWebai · Austin, TX
Full-timeRemote3mo ago
AWSAzureDockerGCPHadoopKubernetes
Cover LetterConnect

Prepare for this interview

Elite

AI-generated questions, company research, and talking points tailored to this role


About the role

webAI is the first end-to-end private AI platform. Enterprises and Governments use webAI to bring AI to their data, powering specialized intelligence trained on their own knowledge and compounding in value with every use. Our approach is guided by a simple philosophy: AI should be specialized, sovereign, efficient, and sustainable. It should solve real problems, not just burn more compute. Our technology enables the development of powerful AI using limited amounts of data, challenging the underlying assumption that big data is the key to unlocking the full power of AI. We are offering a superior career opportunity to join a dynamic and fast-growing team that fosters an exciting and growth-oriented work culture; an opportunity where you can truly put your name on meaningful change for an entire civilization. We are seeking an experienced Staff Machine Learning Engineer with a strong background in Large Language Models (LLMs) and/or Mixture of Experts (MoEs). The ideal candidate will have a proven track record of developing and deploying advanced AI models.

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.

Requirements

  • Advanced degree (Ph.D. preferred) in Computer Science, or a related field.
  • Proven track record of building and 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 Skills:
  • 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.
  • We at webAI are committed to living out the core values we have put in place as the foundation on which we operate as a team. We seek individuals who exemplify the following:
  • Truth - Emphasizing transparency and honesty in every interaction and decision.
  • Ownership - Taking full responsibility for one's actions and decisions, demonstrating commitment to the success of our clients.
  • Tenacity - Persisting in the face of challenges and setbacks, continually striving for excellence and improvement.
  • Humility - Maintaining a respectful and learning-oriented mindset, acknowledging the strengths and contributions of others.

Benefits

Competitive salaryComprehensive health, dental, and vision benefits package401(k) match (U.S.-based employees only)$200/month Health & Wellness stipendContinuing Education support$500/year Function Health subscription (U.S.-based employees only)Free parking for in-office employeesFlexible Time Off (FTO)Parental leave for eligible employeesSupplemental life insuranceHealth insuranceDental insuranceVision insurance401(k)Flexible scheduleParental leave

Your Match

How well this role fits your profile.

Company Intel

What employees say

Worked at webai? Share your experience

Interested in this role?

Apply on the company's website.

Cover LetterConnect