AI Engineer & Researcher - Multimodal

$180,000 - $440,000/Yr

Genai Works - Palo Alto, CA

Full-time - Mid Level

Palo Alto, CA

About the position

The position at xAI involves working as a Member of Technical Staff, focusing on engineering excellence and contributing to the company's mission of creating AI systems that can accurately understand the universe. The role emphasizes hands-on contributions, strong communication skills, and the ability to innovate and implement state-of-the-art methods in deep learning. Engineers are encouraged to work across various areas, fostering a collaborative and flat organizational structure.

Responsibilities

• Train trillion parameter neural networks at scale and various smaller specialized models.
• Implement the latest state-of-the-art methods from deep learning literature rapidly.
• Innovate new ideas to advance the development of AI systems that understand the universe and create new knowledge.
• Build distributed training systems for AI models in HPC clusters.
• Work with JAX/XLA for multi-host training.
• Debug and customize third-party source code.

Requirements

• Experience in building distributed training systems for AI models in HPC clusters.
• Proficiency in JAX/XLA for multi-host training.
• Strong debugging skills, including digging into third-party source code.
• Excellent communication skills to share knowledge concisely and accurately.

Benefits

• Competitive salary range of $180,000 - $440,000 USD.
• Opportunity to work in a highly motivated and innovative team.
• Flat organizational structure promoting initiative and leadership.