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

  1. Scan your resume against this posting to see which of its terms you are missing.
  2. Tailor your resume so the summary and bullets use the posting's own wording where it is true for you.
  3. Read the machine learning engineer resume guide and the interview questions for the role.
  4. 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