Staff Machine Learning Engineer

Summary

Join webAI, a software company building a decentralized AI development platform, as a Staff Machine Learning Engineer. Lead the development and optimization of Large Language Models and Mixture of Experts models. Collaborate with cross-functional teams, conduct cutting-edge research, and mentor junior engineers. This role requires an advanced degree, 5+ years of experience in machine learning with LLM and MoE expertise, and strong programming skills. Preferred skills include cloud computing, big data technologies, and containerization experience. webAI offers competitive salaries, comprehensive benefits, and a flexible work week.

Requirements

• Advanced degree (Ph.D. preferred) in Computer Science, or a related field
• Proven track record of innovations through publications or industry experience
• Minimum of 5 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

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

Preferred Qualifications

• 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

Benefits

• Competitive salary and performance-based incentives
• Comprehensive health, dental, and vision benefits package
• $200/mos Health and Wellness Stipend
• $400/year Continuing Education Credit
• Flexible work week
• Free parking, for in-office employees
• Unlimited Approved PTO
• Parental, Bereavement Leave
• Supplemental Life Insurance