Member of Technical Staff - RL Inference

SpaceXAI is on a mission to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. They are seeking an engineer for their RL infrastructure team to help with low precision RL training and inference.

Responsibilities:

• Design and optimize our inference stack for all shapes of RL workloads at SpaceXAI, from small scale ablations to production training runs
• Analyze, profile and address performance bottlenecks in large scale RL systems
• Work closely with the modelling team to efficiently implement novel RL techniques and algorithms