Quick overview
Job description
We are Genmo, a research lab developing the world’s most sophisticated video world models to understand, simulate, and interact with the physical world. Our mission is to unlock the right brain of AGI. Join us in advancing physical intelligence and enabling robots to learn and act in a changing world.
We're seeking a GPU Performance Engineer to squeeze every last FLOP from our H100 infrastructure and optimize our model serving stack to its absolute limits.
The Role
You'll be our performance optimization expert, using advanced profiling tools to identify bottlenecks and implementing solutions that achieve 5-10x speedups. From writing custom CUDA kernels to eliminating cold start latency, you'll ensure our infrastructure delivers world-class performance. This role is perfect for someone who gets excited about microsecond optimizations and pushing hardware to its theoretical limits.
Key Responsibilities
Profile and optimize GPU workloads using Nsight Systems, nvprof, and custom instrumentation
Write high-performance CUDA and Triton kernels for critical model operations
Optimize cold start latency from seconds to milliseconds for our serving infrastructure
Tune memory access patterns, kernel fusion, and GPU utilization
Collaborate with ML engineers to optimize model implementations
Debug performance issues across the full stack from application to hardware
Implement custom memory pooling and allocation strategies
Share optimization techniques and build performance culture across teams
Qualifications
Bachelor's or Master's degree in Computer Science, Electrical Engineering, or related field
5+ years systems programming experience with 3+ years focused on GPU optimization
Expert proficiency with GPU profiling tools (Nsight Systems, nvprof)
Strong CUDA programming skills with production kernel development
Deep understanding of GPU architecture (memory hierarchy, SMs, warps)
Track record of achieving significant performance improvements (5-10x)
Experience with Python and C++ in production environments
We Value
Experience with Triton kernel development
Knowledge of CUTLASS or similar high-performance libraries
Background in ML-specific optimizations (attention, transformers)
RDMA/InfiniBand optimization experience
Contributions to GPU libraries or frameworks
Low-level debugging skills (PTX/SASS reading)
Genmo is an Equal Opportunity Employer. Candidates are evaluated without regard to age, race, color, religion, sex, disability, national origin, sexual orientation, veteran status, or any other characteristic protected by federal or state law. Genmo, Inc. is an E-Verify company and you may review the Notice of E-Verify Participation and the Right to Work posters in English and Spanish.
Similar jobs
- TT
Senior / Staff Software Engineer, InfrastructureRecently posted Auto Apply
True Talent Solutions Partners · San Francisco · On-site 🇺🇸
DockerAWSCDK+9TechnologySalary not listed - TT
Principal Software Engineer, PerceptionRecently posted Auto Apply
True Talent Solutions Partners · Alameda · On-site 🇺🇸
Edge ComputingRoboticsComputer Vision+3TechnologySalary not listed - TT
Senior Software Engineer, DesignRecently posted Auto Apply
True Talent Solutions Partners · Remote · Remote 🇺🇸
FigmaCSSTailwind CSS+2TechnologySalary not listed - TT
Founding Software EngineerRecently posted Auto Apply
True Talent Solutions Partners · San Francisco · On-site 🇺🇸
LLMPythonReact+1TechnologySalary not listed - TT
Software EngineerRecently posted Auto Apply
True Talent Solutions Partners · New York · On-site 🇺🇸
AWSPythonReact Native+2TechnologySalary not listed - TT
Research Crawling EngineerRecently posted Auto Apply
True Talent Solutions Partners · Remote · Remote 🇺🇸
RustNLPC+++5EngineeringSalary not listed - TT
Lead Product EngineerRecently posted Auto Apply
True Talent Solutions Partners · New York · On-site 🇺🇸
SourcingNext.jsFigma+1Salary not listed - SL

Founding Engineer (DreamLever) - NYRecently posted Auto Apply
Scope Labs · New York City · Hybrid 🇺🇸
EngineeringSalary not listed - CI

Growth EngineerRecently posted Auto Apply
Circleback · San Francisco · On-site 🇺🇸
OnboardingNext.jsSwift+3EngineeringSalary not listed - OF
Application Software Developers/EngineerRecently posted Auto Apply
One Federal Solution · Beltsville, Maryland · On-site 🇺🇸
DockerMicroservicesMySQL+13TechnologySalary not listed - UP

Senior Software Engineer - Machine Learning PlatformRecently posted Auto Apply
Upstart · United States | Remote · Remote 🇺🇸
Apache SparkAWSMLOps+7Technology$166.9k/yr - TO
Staff Software Engineer, OrdersRecently posted Auto Apply
Toast · Remote · Remote 🇺🇸
MicroservicesDiversity & InclusionGraphQL+4TechnologySalary not listed
