INACTIVE
Full-Time
AI Engineer & Researcher
Posted on 4/18/2024
Develops AI for scientific discovery
Compensation Overview
$180,000 - $440,000Annually
Expert
Palo Alto, CA, USA + 1 more
Required Skills
Python
CUDA
Requirements
- - Strong experience with Rust, Python, JAX, XLA, and CUDA (C++ and Triton)
- - Experience in training trillion parameter neural networks at scale
- - Proficiency in mastering all data modalities: text, audio, images, video, etc
- - Ability to profile, debug, and optimize multi-host GPU utilization
- - Familiarity with rapidly implementing the latest state of the art methods from the deep learning literature
- - Experience in building distributed training systems for AI models in HPC clusters
- - Proficiency in configuring and troubleshooting hardware and operating-systems for maximum performance
- - Familiarity with JAX/XLA for multi-host training
- - Ability to write custom CUDA kernels in either C++ or via Triton
- - Experience in digging into third-party source code for debugging and customization
Responsibilities
- - 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
- - Digging into third-party source code for debugging and customization
xAI is dedicated to harnessing advanced AI algorithms to enhance human scientific discovery, particularly in expanding our understanding of the universe. Working here provides a unique opportunity to be at the forefront of AI-driven research with a team that holds deep expertise in the field. The culture fosters collaboration and continuous learning, making it an ideal environment for professional growth and innovative pursuits in the realm of science and technology.
Company Stage
M&A
Total Funding
$6.1B
Headquarters
New York, New York
Founded
2023
Growth & Insights
Headcount
6 month growth
↑ 0%1 year growth
↑ 0%2 year growth
↑ 0%INACTIVE