AI Infrastructure Software Engineer — CosmosLab

This role involves designing and improving large-scale AI training infrastructure for NVIDIA's Physical AI world foundation models. Key duties include developing pre-training and supervised fine-tuning pipelines, optimizing inference stacks, and integrating simulation environments for reinforcement learning. The position requires deep expertise in distributed systems, GPU optimization, and software engineering practices to ensure high throughput and scalability.