Member of Technical Staff - RL Training Framework
$180kxAI
Job Description
Job Description
ABOUT xAI
xAI's mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for individuals who appreciate challenging themselves and thrive on curiosity. We operate with a flat organizational structure. All employees are expected to be hands-on and to contribute directly to the company's mission. Leadership is given to those who show initiative and consistently deliver excellence. Work ethic and strong prioritization skills are important. All employees are expected to have strong communication skills. They should be able to concisely and accurately share knowledge with their teammates.
ABOUT THE ROLE:The RL infrastructure team is looking for an engineer to help develop our RL training framework.
RESPONSIBILITIES:
- Design and implement the systems backing all RL workloads at xAI, from small scale ablations to production training runs.
- Profile, debug, and optimize end-to-end training performance
- Improve scalability and observability of the RL stack
- Experience building, debugging, and optimizing efficiency of large-scale distributed systems
- Comfortable diving into unfamiliar areas and solving problems at all levels of the stack
- Proficiency in Python, Jax, Rust, and/or C++
- Experience with large scale LLM training infrastructure
- Strong knowledge of reinforcement learning techniques
- Experience with RL numerics
$180,000 - $440,000 USD
Base salary is just one part of our total rewards package at xAI, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks.
xAI is an equal opportunity employer. For details on data processing, view our Recruitment Privacy Notice.
Vacancy posted 18 days ago
Similar jobs that could be interesting for youBased on the Member of Technical Staff - RL Training Framework in Palo Alto, CA vacancy
- ...applied AI research lab pioneering data and RL environment curation for training and evaluating agents. Recently, we... ...rollouts and identifying subtle patterns. Technical execution: Proficiency in Python and ML frameworks (PyTorch, JAX, or similar). Experience with...TrainingFlexible hours
- ABOUT THE ROLE: You will work on the most critical post-training and reinforcement learning challenges at any given time — including reward modeling, preference optimization (RLHF/DPO), and RL for improving reasoning, truthfulness, and real-world capabilities. You will...Training
- About the Role As a Member of Technical Staff [Research] at NeoCognition , you’ll... ...of LLM reasoning, post-training, and agentic system design.... ...training (instruction tuning, RL, reasoning) Data pipeline... ...familiarity with modern ML frameworks (e.g., PyTorch, JAX, or TensorFlow...Training
- Member of Technical Staff — Kernel / Compiler / Communication About the Role RadixArk is seeking a Member... .... This role is critical to scaling training and inference across thousands of... ...and developed Miles, our large‑scale RL framework. We build world‑class systems for AI...TrainingFlexible hours
- Member of Technical Staff Physical AI (Robotics / World Models) Palo Alto, CA About Orbifold... ...of data, evaluation, and model training. We design evaluation harnesses that... ...-designing datasets, evaluation frameworks, and training and RL pipelines that shape how their...TrainingShift work
- RadixArk is seeking a Member of Technical Staff — Training to build and scale the systems that train frontier... ...running training jobs Develop training frameworks and infrastructure tooling... ...and developed Miles, our large-scale RL framework. We build world-class infrastructure...TrainingFlexible hours
$180k - $250k
Member of Technical Staff -- TPU Systems (JAX / XLA / PALLAS) About the Role RadixArk... ...-performance inference and training systems using JAX, XLA, and... ...JAX, XLA, or TPU-focused frameworks Bachelor's or Master's... ...developed Miles (our large-scale RL framework). We're on a...TrainingFull timeFlexible hours- Member of Technical Staff — Supercomputing About the Role RadixArk is hiring a Member of Technical Staff... ...for frontier‑scale inference and training workloads. This role sits at the intersection... ...developed Miles , our large‑scale RL framework. We are building world‑class open...TrainingFlexible hours
$180k
...About the Role The mid‑training team at xAI aims to provide... ...boost the ceiling for RL. Engineer long‑context... ...Spark, Ray, and other frameworks for large‑scale data... ...interview”) during which a member of our team will ask... ...which consists of four technical interviews: Coding assessment...TrainingTemporary workRelocation$180k
Member of Technical Staff - Multimodal Understanding About xAI xAI’s mission is... ...curation/acquisition, tokenizer training, large‑scale pre‑training,... ...models. Create evaluation frameworks, internal benchmarks,... ...stack (pre‑training > SFT/RL/post‑training) to enable reasoning...TrainingTemporary work$180k
...reinforcement learning data for training Grok reasoning models. Our... ...training data, and advancing RL algorithms. About the Role... ...phone interview”) during which a member of our team will ask some basic... ...process, which consists of four technical interviews: # Coding...TrainingTemporary workWork at officeWork from homeRelocation- RadixArk is hiring a Member of Technical Staff — CI Engineer to own the infrastructure that keeps SGLang... ...and developed Miles (our large-scale RL framework). We’re on a mission to democratize... ...class open systems for inference and training. Our team has optimized kernels...TrainingFlexible hoursNight shift
$324k - $396k
...share knowledge with their teammates. Member of Technical Staff (X.AI LLC; Palo Alto, CA): Build... ...include models and large language model training, pipeline parallelism training of... ...languages (such as C++, Java), using ML frameworks (such as PyTorch, TensorFlow, or JAX)...TrainingXai
Palo Alto, CA11 days ago- ...Radiology Partners, Cognita’s models are trained and validated on one of the world’s... ...datasets. About the Role As a Member of Technical Staff on the ML Infrastructure team, you will... ...industry-standard distributed computing frameworks. Experience deploying large language...Training
- ...Radiology Partners, Cognita’s models are trained and validated on one of the world’s... ...datasets. About the Role As a Member of Technical Staff focused on ML Evaluation, you will be... ...and blind spots. Develop evaluation frameworks and methodologies for multimodal and...Training
- ...Reinforcement Learning experiments (GRPO/PPO/DPO), training data mixes and reward signal explorations... ...are also encouraged to apply. RL Knowledge: Strong academic understanding... ...proficiency in Python and deep learning frameworks (PyTorch). You should be able to write clean...TrainingInternship
$180k
Member of Technical Staff - Imagine Safety ABOUT xAI xAI’s mission is to create AI systems that can... ...safeguards, detection systems, and evaluation frameworks that prevent harm while preserving... ...interactions, model outputs, and training data to continuously improve safety...TrainingTemporary workWorldwide$180k
Member of Technical Staff, Recommendation Systems About xAI xAI’s mission is to create AI systems that... ...experience Write data pipelines and training jobs that continuously learn from... ...scale Skilled in one or more DL software frameworks such as JAX or PyTorch Exceptional...TrainingTemporary workRelocation$148.5k - $223.9k
Senior Member of Technical Staff - AI ResearchSkip to main content#Senior Member of Technical Staff... ...Hands-on experience with deep learning frameworks** *Strong understanding of ML... ...Experience implementing and debugging model training, evaluation, and inference pipelines*...TrainingWork at office$200k - $300k
...This team works on foundational model capabilities, post-training techniques, building RL infra and infrastructure that benefits the entire... ...model development Build robust and effective training frameworks (on top of Megatron/PyTorch) for post-training LLMs Implement...TrainingFull time$180k
...Responsibilities span data curation, modeling, training, inference serving, and product... ...long‑horizon synthesis, agentic planning, RL training, and world simulation (including... ...visual and audio data. Design evaluation frameworks, metrics, benchmarks, evals, and reward models...TrainingTemporary work- ...real robot tasks. Post-training at Rhoda means taking a... ...levels — from senior to staff. What You'll Do Design and implement RL training pipelines to improve... ...Build evaluation frameworks for post-trained policies... ...are expected to define technical direction and drive research...TrainingShift work
$324k - $396k
...About the Role Member of Technical Staff (X.AI LLC; Palo Alto, CA): Introduce innovative techniques and analyses to the AI field to facilitate... ...and language understanding. Stabilize large language model training, pipeline parallelism training of large language models,...TrainingRemote workXai
Palo Alto, CA1 day ago$150k
...Member of Technical Staff Location: Palo Alto, CA Company Stage of Funding: Seed Stage AI Startup ($27M Raised) Office Type: Onsite... ...trading firms. Background in AI model evaluation, post-training, reinforcement learning, or large language model...TrainingWork at officeVisa sponsorshipRelocation packageRecruiting from Scratch
Palo Alto, CA4 days ago$175k - $350k
...value for all. About the Role As a Model Training engineer, you will design, build, and scale... ...to ship. Communicate crisply with both technical and non-technical teammates. Have a bachelor... ...alignment techniques (e.g., curriculum RL, multi-objective reward modeling, tool-...TrainingInflection AI
Palo Alto, CA5 days ago- Kindredventures in Palo Alto is looking for a Founding Member of the Technical Staff to lead AI model training for chip design. You will work on cutting-edge Reinforcement Learning environments and build end-to-end ML pipelines. The successful candidate will hold a PhD...Training
$324k - $396k
Pantera Capital is seeking a Member of Technical Staff in Palo Alto, CA. This role involves introducing innovative techniques in AI, stabilizing large language model training, and working on hands-on technical problems. Candidates should have relevant experience and a Bachelor...TrainingRemote job- ...reliable despite non-deterministic model behavior. Role As a Member of Technical Staff, Machine Learning, you will build core ML components. You... ...ML. Focus Build and improve ML components across data, training, evaluation, and inference. Fine-tune and adapt models as...Training
- ...iterate agentic AI systems with customers. With your strong technical competence, strategic thinking, and customer engagement, you... ...meaningful implementation track records, such as LLM, pre/post‑training, RL, agentic system. Prioritizes deep and strategic thinking....Training
- ...SuperIntelligence, xAI, Apple and Intel. What You’ll Do As a Founding Member of the Technical Staff (Applied AI) at Architect, you’ll sit at the intersection... ...ones. Partner closely with the ML research, post‑training, and infra teams to turn hardware domain expertise into...Training
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Member of Technical Staff - RL Training Framework. Be the first to apply!
Related searches
- infrastructure support analyst Palo Alto, CA
- systems support technician Palo Alto, CA
- application support technician Palo Alto, CA
- desktop support analyst Palo Alto, CA
- tech aide Palo Alto, CA
- support technician Palo Alto, CA
- product support technician Palo Alto, CA
- helpdesk support technician Palo Alto, CA
- technical associate Palo Alto, CA
- help desk assistant Palo Alto, CA