Pytorch Engineer
Core
Designing, implementing, optimizing, and maintaining PyTorch and Triton software stacks for Graphcore's AI hardware accelerators.
Role type
Senior IC machine-learning framework engineer (PyTorch/Triton)
Builds
ML framework integrations and optimizations for Graphcore IPU hardware
Domain
AI hardware / Machine Learning Frameworks
Deliverable
production ML models | product features
Required skills
Python development, C++ application development, ML kernel development, compiler development, ML framework development
Preferred skills
Open-source community collaboration, technical documentation, code review, mentoring
Technologies
PyTorch, Triton, C++, Python
Responsibilities
Designing and implementing new features, optimizing software stack performance, managing code quality and technical debt, conducting code reviews, resolving regressions and defects, contributing to upstream open-source projects
Seniority
Senior, hands-on IC
