Member of Technical Staff - RL Training Framework
$180kSpaceXAI
Job Description
Job Description
SpaceX was founded under the belief that a future where humanity is out exploring the stars is fundamentally more exciting than one where we are not. Today, SpaceX is actively developing the technologies to make this possible, with the ultimate goal of making life multiplanetary and expanding consciousness to the stars.
SpaceXSI is building Super Intelligence to understand the universe and help expand humanity's knowledge and capabilities. We work across the full Super Intelligence stack: compute (Colossus supercomputers), models (Grok), and applications (Grok Bot, Grok Build, Grok Imagine, Grok Voice, and X). We're creating a powerful flywheel where more compute trains better models, better models power more useful products, and what we learn from people using those products shapes what we build next.
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 SpaceXAI, 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 SpaceXAI, 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.
SpaceXAI is an equal opportunity employer. For details on data processing, view our Recruitment Privacy Notice.
Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Member of Technical Staff - RL Training Framework in Palo Alto, CA vacancy
$180k
...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...TrainingTemporary workSpaceXAI
Palo Alto, CAa month ago$180k - $230k
...Job Description Job Description Member of Technical Staff, ML Systems Company: TensorScale... ...TensorScale TensorScale builds the training and inference stack for world models.... ...features in an inference or training framework (vLLM, SGLang, TensorRT-LLM, Megatron...TrainingFull timeH1bWork at officeVisa sponsorshipTransparent Search Group
Menlo Park, CA19 days ago$180k
...through synthetic data generation. Optimize mid-training data mixtures to boost the ceiling for RL. Engineer long-context data recipes. Develop robust... ...engineering abilities in Spark, Ray, and other frameworks for large-scale data processing. COMPENSATION AND...TrainingTemporary workSpaceXAI
Palo Alto, CAa month ago$180k
...: data curation/acquisition, tokenizer training, large-scale pre-training, post- training... ...trillion-parameter models. Create evaluation frameworks, internal benchmarks, reward models, and... ...across the stack (pre-training → SFT/RL/post-training) to enable reasoning, tool...TrainingTemporary workSpaceXAI
Palo Alto, CAa month ago$280k - $350k
...relocate. About this role As a Member of Technical Staff [Research] at NeoCognition, you'll be... ...research initiatives in LLM reasoning, post-training, and agentic system design Develop... ...post-training (instruction tuning, RL, reasoning), RLHF and open-weight models...TrainingFull timeRemote workRelocationVisa sponsorshipRelocation package$180k
...teammates. 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...TrainingTemporary work$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 workSpaceXAI
Palo Alto, CAa month 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, CA5 days ago$180k
...accurately share knowledge with their teammates. ABOUT THE ROLE: The RL infrastructure team is looking for an engineer to help develop... ...signal quality, and building internal tooling to support training our next-generation of models Identify and resolve performance...TrainingTemporary workSpaceXAI
Palo Alto, CA11 days ago$180k
...design and scale safeguards, detection systems, and evaluation frameworks that prevent harm while preserving creativity and user freedom... ...feedback loops between user interactions, model outputs, and training data to continuously improve safety while maintaining high performance...TrainingTemporary workWorldwideSpaceXAI
Palo Alto, CAa month ago$180k - $230k
...the Role Join an ML systems team building and improving the training and inference stack for image, video, and world-model workloads... ...Experience contributing core features to an inference or training framework, rather than focusing only on deployment or integration....TrainingFull timeVisa sponsorship$180k - $230k
...About the Role Join an AI infrastructure team building efficient training and inference systems for image, video, and world-model... ...Nsight. Authorship of core features in an inference or training framework, such as vLLM, SGLang, TensorRT-LLM, or Megatron. Deployment...TrainingFull timeVisa sponsorship$180k - $280k
...type of firm, where live capital is the training ground for an intelligence that compounds... ...AI is used in investing and shape the technical direction of our agent platform from an... ...Working at Poesis As an early team member, you’ll help shape not just the product,...TrainingFull timeWork at officeVisa sponsorshipWork visaRelocation package3 days per week$200k - $280k
...re building a new type of firm, where live capital is the training ground for an intelligence that compounds with every signal... ...and intelligent risk management. We’re looking for a Member of Technical Staff (Machine Learning) to build the models, signals, and evaluation...TrainingFull timeWork at officeVisa sponsorshipWork visaRelocation package3 days per week$200k - $300k
...open to transfers (OPT, H1B). About this role As a Member of Technical Staff [Product/Agent] at NeoCognition, you'll be one of the founding... ...), backend serving layer (platform), ML/LM research (training, data, evals). This role sits in the agent harness/product...TrainingFull timeH1b$180k - $280k
...We're building a new type of firm, where live capital is the training ground for an intelligence that compounds with every signal. About the Role We’re looking for an exceptional Member of Technical Staff (Quantitative Finance) to work alongside our CEO, Chief...TrainingFull timeWork at officeVisa sponsorshipWork visaRelocation package3 days per week$180k - $220k
...improving APIs to enable more complex engineering workflows Set up and integrate LLM or VLM infrastructure Set up an MLOps framework for training deep learning architectures for geometry and physics data Assist in the strategy, planning the product roadmap, and...TrainingFull time$200k - $240k
...backed by Khosla Ventures and Eclipse Ventures. The role Training a foundation model for physics means holding petabytes of... ...recurring question lives somewhere other than in your head. Technical direction. Work with the AI team on what the next model will...TrainingFull time$270k - $350k
...labs on their most consequential data, evaluation, and post-training challenges, building the systems that turn expert human... ...knowledge useful for advancing AI. The Role We are hiring a Member of Technical Staff to help define how frontier coding agents are benchmarked...TrainingFull timeWork at officeFlexible hours$264k - $342k
...Technical Staff-Storage Architect From applied research to advanced engineering... ...stacks—from software frameworks through compute, storage,... ...Every Dell Technologies team member brings something unique to the... ...workflows, data pipelines, training/inference patterns, data...TrainingDell
Santa Clara, CA2 days ago- ...services/APIs) Own projects full lifecycle Design discussions & technical scoping Implementation & testing Post-launch iteration... ...in React and some level of experience with one backend framework (e.g., Node.js, Django, or similar). Some familiarity with databases...Work at officeRemote workVisa sponsorshipMonday to Friday
ProductNow
Palo Alto, CAa month ago $230k - $350k
...Job Description Job Description Member of Technical Staff, Inference Systems Company: Photon Location: Palo Alto, CA (on-site 5 days per week) Compensation: $230,000 - $350,000 + competitive equity Employment Type: Full-time Visa Sponsorship: Visa...Full timeH1bVisa sponsorshipTransparent Search Group
Palo Alto, CA19 days ago- ...expansion will include a team of Technical Support Specialists. We are... ...an equivalent combination of training and experience is desirable.... ...in daily workflow; KCS framework experience is an added advantageAbility... ...an environment where team members support each other, share...TrainingWork at officeWorldwide
Fortinet
Sunnyvale, CA20 hours ago - ...strong opinions about how systems should work, grounded in data and deep understanding of reality. Every engineer here is deeply technical and a relentless problem-solver. We think deeply, not frantically. How We Work Our culture is built on the premise that...Full time
Whop
Palo Alto, CA20 hours ago - ...capabilities into simple user experiences. Establish frontend architecture patterns, maintain testing standards, and mentor team members. Skills and qualifications: Experience building production web applications in React and TypeScript Deep understanding of...
Coram AI
Sunnyvale, CA1 day ago - ...Job Title: Member of Technical Staff Location - Santa Clara, CA 95054 Duration: 6 Months Job Description: Full life cycle software development from design through development, validation and release. Implementation of software for inventory, configuration...
Global Pharma Tek
Santa Clara, CA2 days ago $184k - $287.5k
Reinforcement learning post-training is driving some of the most significant... ...challenges in the field. RL requires inference, rollout... ...do.NVIDIA is building an RL Frameworks engineering team to develop... ...one or more of the following technical areas:Reinforcement learning...TrainingFull timeNvidia
Santa Clara, CA2 days ago- Job Title: Senior Technical Support SpecialistFortinet, a cybersecurity... ...an equivalent combination of training and experience is desirable.... ...in daily workflow; KCS® framework experience is an added advantage... ...fostering an environment where team members support each other, share...TrainingWork at officeWorldwide
- ...Respond to product functionality and technical support requests for our enterprise business... ...Self-starter, working as hybrid team member Willing to help provide input and insight... ...of absence, compensation, benefits, training, and apprenticeship. CENTRL makes hiring...TrainingTemporary workApprenticeshipWork at officeLocal areaWorldwideDay shift3 days per week
CENTRL
Mountain View, CA3 days ago $272k - $431.25k
...Reinforcement learning post-training is where modern AI systems learn... ...problems in AI: a single RL run ties together inference,... ...challenge, NVIDIA is building an RL Frameworks engineering team for the open... ...team, and convert emerging technical, customer, and partner...TrainingFull timeRemote workNvidia
Santa Clara, CA1 day ago
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
- product support technician Palo Alto, CA
- mri tech aide Palo Alto, CA
- IT assistant Palo Alto, CA
- IT help desk technician Palo Alto, CA
- systems support technician Palo Alto, CA
- help desk technical support Palo Alto, CA
- technical analyst Palo Alto, CA
- tech assistant Palo Alto, CA
- IT support technician Palo Alto, CA
- desktop support analyst Palo Alto, CA