Member Of Technical Staff (AI)

TL;DR

Member of Technical Staff (AI): Developing and stabilizing large-scale language model training pipelines with an accent on quantitative reasoning and language understanding. Focus on implementing advanced neural network architectures, optimizing distributed training systems, and deploying cutting-edge deep learning models.

Location: Must be based in or able to work from Palo Alto, CA (Telecommuting permitted within the US)

Salary: $324,000 - $396,000 per year

Company

xAI is an AI research company dedicated to creating systems that accurately understand the universe and aid humanity in its pursuit of knowledge.

What you will do

• Stabilize large language model training and pipeline parallelism.
• Implement innovative techniques for quantitative reasoning and language understanding.
• Design and build scalable data processing systems using big data technologies.
• Deploy large-scale deep learning systems on hardware accelerators like GPUs and TPUs.
• Collaborate with researchers to develop applications in NLP, computer vision, and speech recognition.
• Perform hands-on implementation of advanced neural network architectures.

Requirements

• Bachelor’s degree or foreign equivalent in Computer Science, Engineering, or related field.
• 2 years of professional experience in the job offered or a computer-related occupation.
• Experience with big data systems such as Spark, Hadoop, or BigQuery.
• Proficiency in building large-scale Kubernetes clusters for data storage and processing.
• Strong programming skills in Rust, C++, or Python.
• Experience building applications with hardware accelerators (GPUs/TPUs) on cloud platforms.

Culture & Benefits

• Flat organizational structure promoting direct contribution.
• High-performance environment focused on engineering excellence.
• Opportunity to work on cutting-edge AI research and large-scale infrastructure.
• Emphasis on strong communication and initiative-driven leadership.