Be at the forefront of the AI revolution by joining a leading AI lab''s GenAI team, where your expertise will drive the development of advanced Large Language Models. This role offers a unique opportunity to build foundational AI models from the ground up, focusing on modern ML frameworks such as JAX and PyTorch, along with kernel-level programming.
Key Responsibilities - Guide research and engineering teams to close knowledge gaps and enhance AI model performance in MLOps, training infrastructure, and ML framework-level topics.
- Design challenging, domain-relevant tasks, and create accurate and well-structured solutions to MLOps and ML systems problems.
- Evaluate MLOps tasks and solutions, providing clear, written technical feedback.
- Develop guidelines and detailed rubrics/evaluation frameworks to assess training pipeline design, distributed systems reasoning, and kernel-level optimization across tasks.
- Collaborate with o...