Reinforcement Learning Engineer at SpaceXAI

Join SpaceXAI as a Reinforcement Learning Engineer, contributing to our ambitious mission of developing AI systems for deep understanding of the universe. Work hands-on to enhance our RL training framework through innovative engineering solutions.

As part of the RL infrastructure team, you will implement systems that support RL workloads from initial testing to large-scale training runs. Candidates should demonstrate a strong background in building and optimizing distributed systems, as well as proficiency in languages like Python and C++. Your capacity for problem-solving and communication will be vital in this collaborative environment.

Key Responsibilities:

• Design systems for RL workloads and training runs

• Profile and optimize end-to-end training performance

• Increase scalability and observability of RL stacks

Requirements:

• Experience with large-scale distributed systems

• Strong proficiency in Python, Jax, Rust, or C++

• Ability to tackle challenges across all stack levels

Become a key player in advancing AI technology at SpaceXAI through hands-on development and collaborative engineering.
#J-18808-Ljbffr