Software Engineer, Triton Compiler
OpenAI
over 4 years ago
San Francisco, CA, USA
Mid Level / Senior
Base Salary
$266k - $445k/yr
Responsibilities
- Design and implement state-of-the-art AI models.
- Optimize machine learning code for performance and reliability.
- Develop and enhance the Triton compiler for GPU kernels.
- Collaborate with teams to accelerate the development of AI capabilities.
- Identify and solve meaningful problems to advance research productivity.
Requirements
- 3+ years of relevant engineering experience.
- Experience in owning problems end-to-end.
- Willingness to learn and acquire necessary knowledge for project completion.
- Bonus: Contributions to AI frameworks like PyTorch or TensorFlow, or compilers like GCC, LLVM, or MLIR.
Tech Stack
PythonPyTorchTensorFlow
Categories
AI & MLBackend