webAI is the first end-to-end private AI platform, enabling enterprises and governments to utilize AI for their data. The Staff Machine Learning Engineer will lead the development of advanced AI models, collaborate with cross-functional teams, and conduct research to enhance model performance and efficiency.
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
- 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