The ML Fundamentals team develops the core tooling to empower ML engineers to operate faster and more reliably. The ML Fundamentals team is responsible for the key building blocks that power ML training, model acceleration, model deployment, and ML overall developer experience. As the ML Fundamentals TLM, you will drive the vision and lead development of the key tools that power ML development at Stack.
Job listings
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. You will work with modern numerical computing and deep learning technology, contribute to open-source projects like PyTorch, TorchRL, TorchAO, and Triton, and use C++, CUDA, Python, C, and MLIR.