- Productize the software stack for the AI compute engine.
- Develop, enhance, and maintain software kernels for ML operators including softmax, layer norm, and activation functions.
- Optimize kernels for next-generation AI hardware.
- Build solutions to ensure the SDK is intuitive for developer use.
- Analyze performance and navigate hardware-software co-design trade-offs.
- Deliver and scale high-quality software within a fast-moving development environment.
PythonMachine LearningPyTorch+3 more