We are looking for an enthusiastic and highly motivated software engineer with experience and interest in numerical and high-performance computing to join our PyTorch team. In this position, you’ll get to work with modern numerical computing and deep learning technology and contribute to open-source projects.
Job listings
In this role, you’ll architect and build scalable cycle-accurate AI accelerator performance models that can run representative AI workloads. Your work will directly inform how we design and optimize future hardware platforms, from high-level performance models to cycle-accurate simulations. This role is remote, based out of North America or Hybrid from one of our offices.
The Auto Labeling team at Stack develops large ML models to generate high-quality labeled data, essential for training and evaluating Stack's onboard perception models. The models leverage state-of-the-art multi-sensor fusion techniques, integrating data from lidars, cameras, radars, and IMUs. Significant design contributions and collaboration with onboard and offboard consumers of labeled data are expected.