Inference at Genesis AI
Bay AreaEngineering & ResearchFull-timePosted 6 months ago
About the role
As published on Genesis AI's job board.
WHAT YOU’LL DO
- Build low-latency inference pipelines for on-device deployment, enabling real-time next-token and diffusion-based control loops in robotics
- Design and optimize distributed inference systems on GPU clusters, pushing throughput with large-batch serving and efficient resource utilization
- Implement efficient low-level code (CUDA, Triton, custom kernels) and integrate it seamlessly into high-level frameworks
- Optimize workloads for both throughput (batching, scheduling, quantization) and latency (caching, memory management, graph compilation)
- Develop monitoring and debugging tools to guarantee reliability, determinism, and rapid diagnosis of regressions across both stacks
WHAT YOU’LL BRING
- Deep experience in distributed systems, ML infrastructure, or high-performance serving (8+ years)
- Production-grade expertise in Python, with strong background in systems languages (C++/Rust/Go)
- Low-level performance mastery: CUDA, Triton, kernel optimization, quantization, memory and compute scheduling
- Proven track record scaling inference workloads in both throughput-oriented cluster environments and latency-critical on-device deployments
- System-level mindset with a history of tuning hardware–software interactions for maximum efficiency, throughput, and responsiveness
Before you apply
- Scan your resume against this posting to see which of its terms you are missing.
- Tailor your resume so the summary and bullets use the posting's own wording where it is true for you.
- Read the machine learning engineer resume guide and the interview questions for the role.
- Apply on Genesis AI's site. Applications never pass through OneTwo Resume.
About Genesis AI
Full-stack general-purpose robot company that develops the Genesis physics simulator and robot foundation models.
All open roles at Genesis AI · All machine learning and AI engineering jobs at AI startups
More roles at Genesis AI
Product Marketing Lead
Bay AreaProduct & Marketing Posted 5 days ago
Social Media Manager
Bay AreaProduct & Marketing Posted 5 days ago
Head of Supply Chain (Robot Production)
Bay AreaManufacturing Posted 6 days ago
Deployment Program and Operation Manager
Bay AreaDeployment Posted 23 days ago
Deployment Solution Engineer
Bay AreaDeployment Posted 27 days ago