x.ai
Acquired by Bizzabo
Artificial Intelligence • Machine Learning • Natural Language Processing • Invisible Software • vertical AI
201 - 500
AI Engineer & Researcher - Pre-training
May 22
🏢 In-office - San Francisco
x.ai
Acquired by Bizzabo
Artificial Intelligence • Machine Learning • Natural Language Processing • Invisible Software • vertical AI
201 - 500
Description
• Training trillion parameter neural networks at scale, as well as a variety of smaller specialized models. • Profiling, debugging, and optimizing multi-host GPU utilization. • Improving and innovating inference stack to serve millions of users. • Rapidly implementing the latest state-of-the-art methods from the deep learning literature. • Innovating new ideas to develop AI systems that accurately understand the universe and create new knowledge.
Requirements
• Building distributed training systems for AI models in HPC clusters. • Configuring and troubleshooting hardware and operating-systems for maximum performance. • Working with JAX/XLA for multi-host training. • Writing custom CUDA kernels in either C++ or via Triton.
Benefits
• California Consumer Privacy Act (CCPA) Notice
Apply Now