About the role
What You'll Do
Lead the evolution of our high-performance robotics simulation platform
Design and implement the compute infrastructure and data flow mechanisms to optimize performance for physics simulation and foundation model training
Lead development of our compiler stack, focusing on JIT compilation, LLVM IR, and GPU codegen to minimize compile time and maximize runtime performance
Collaborate with the team to improve the compiler's support for differentiable programming, crucial for training neural networks within simulations
Stay current on state-of-the-art ML compilers—such as those in torch, Triton, and JAX—and decide which techniques and approaches are best suited for our application
Work closely with simulation and robotics engineers to align compiler enhancements with application needs
Contribute to relevant open-source projects and participate actively in the broader compiler and systems community
What You’ll Bring
Strong background in compiler construction, particularly in JIT compilation and LLVM-based code generation
Extensive experience with GPU programming models (e.g., CUDA, Vulkan) and understanding of GPU architecture
Track record as a core contributor to GPU programming infrastructure—such as Torch, JAX, Mojo, Taichi, or Warp
Proven ability to profile and optimize complex systems for performance and scalability
Understanding of automatic differentiation and its application in simulation and machine learning contexts
Excellent communication skills and a collaborative approach to problem-solving
Enthusiasm for contributing to and engaging with open-source communities
Skills & Tags
Aplyr's read
Genesis AI is a cutting-edge AI company focusing on foundational and applied AI technologies, attracting top technical talent for its global operations.
What's promising
- •Genesis AI is at the forefront of AI innovation, specializing in foundational models and robotics.
- •The company offers diverse roles across global tech hubs, indicating robust growth and international expansion.
- •Genesis AI's focus on foundational models suggests a commitment to long-term technological leadership.
What to watch
- •Limited public information about company culture and employee satisfaction could concern potential applicants.
- •High concentration of technical roles may limit opportunities for non-technical professionals.
- •The competitive AI industry landscape poses challenges for sustained differentiation and market share.
Why Genesis AI
- •Genesis AI's emphasis on foundational models and robotics sets it apart in the AI sector.
- •The company has a strong international presence, with roles in major tech cities like Paris, London, and the Bay Area.
- •Genesis AI's focus on both foundational and applied AI technologies provides diverse career pathways.
Aplyr’s read is generated by AI from public sources. Was it useful?
Similar roles
Member of Technical Staff (ML Engineer, Recommendations & User Modeling)
Perplexity AI
Senior Member of Technical Staff: ML Systems and Infrastructure
DevRev
Member of Technical Staff, Data & ML Infrastructure for Video Models
Cantina
Member of Technical Staff - Applied ML, RecSys
Liquid AI
Member of Technical Staff - ML Systems & Inference
Gimlet Labs
Member of Technical Staff - ML Engineer / Scientist (JP Localization)
Liquid AI