Genesis AI · 4 hours ago
Staff Software Engineer, Training (Bay Area / Paris / Remote)
United States
Full-time
Remote
Lead/Staff
8+ years expGenesis AI is a company focused on advanced artificial intelligence solutions. They are seeking a Staff Software Engineer to drive improvements in model training efficiency and optimize distributed training systems.
Artificial Intelligence (AI)Robotics
Responsibilities
Drive down wall-clock time to convergence by profiling and eliminating bottlenecks across the foundation model training stack stack, from data pipelines to GPU kernels
Design, build, and optimize distributed training systems (PyTorch) for multi-node GPU clusters, ensuring scalability, robustness, and high utilization
Implement efficient low-level code (CUDA, cuDNN, Triton, custom kernels) and integrate it seamlessly into high-level training frameworks
Optimize workloads for hardware efficiency: CPU/GPU compute balance, memory management, data throughput, and networking
Develop monitoring and debugging tools for large-scale runs, enabling rapid diagnosis of performance regressions and failures
Qualification
Required
Deep experience in distributed systems, ML infrastructure, or high-performance computing (8+ years)
Production-grade expertise in Python
Low-level performance mastery: CUDA/cuDNN/Triton, CPU–GPU interactions, data movement, and kernel optimization
Scaling at the frontier: experience with PyTorch and training jobs using data, context, pipeline, and model parallelism
System-level mindset with a track record of tuning hardware–software interactions for maximum utilization
Company
Genesis AI
Genesis AI operates as a physical AI lab and full-stack robotics company.
11-50 employees
Funding
Current Stage
Early StageTotal Funding
$105M2025-07-01Seed· $105M
2025-06-27Debt Financing
Leadership Team
Theophile Gervet
Co-Founder
Company data provided by crunchbase