echojobs.io
Login

Genesis AI

Member of Technical Staff, ML Compiler and Systems

Remote Bay Area
Python PyTorch Machine Learning
Search for More Jobs
Description

Member of Technical Staff, ML Compiler and Systems (Bay Area, Remote)

Department: Workload

Location: Bay Area, Remote

Employment Type: FullTime

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

Genesis AI
Genesis AI

0 applies

0 views

There are more than 50,000 engineering jobs:

Subscribe to membership and unlock all jobs

Engineering Jobs

60,000+ jobs from 4,500+ well-funded companies

Updated Daily

New jobs are added every day as companies post them

Refined Search

Use filters like skill, location, etc to narrow results

Become a member

🥳🥳🥳 452 happy customers and counting...

Overall, over 80% of customers chose to renew their subscriptions after the initial sign-up.

To try it out

For active job seekers

For those who are passive looking

Cancel anytime

Frequently Asked Questions

  • We prioritize job seekers as our customers, unlike bigger job sites, by charging a small fee to provide them with curated access to the best companies and up-to-date jobs. This focus allows us to deliver a more personalized and effective job search experience.
  • We've got over 200,000 jobs from 15,000+ vetted companies. No fake or sleazy jobs here!
  • We aggregate jobs from 15,000+ companies' career pages, so you can be sure that you're getting the most up-to-date and relevant jobs.
  • We're the only job board *for* software engineers, *by* software engineers… in case you needed a reminder! We add thousands of new jobs daily and offer powerful search filters just for you. 🛠️
  • Every single hour! We add 2,000-3,000 new jobs daily, so you'll always have fresh opportunities. 🚀
  • Typically, job searches take 3-6 months. EchoJobs helps you spend more time applying and less time hunting. 🎯
  • Check daily! We're always updating with new jobs. Set up job alerts for even quicker access. 📅

What Fellow Engineers Say