PyTorch Engineer
Core
Designing, implementing, optimizing, and maintaining PyTorch and Triton software stacks for Graphcore's machine learning accelerators to support ML engineers and researchers.
Role type
Senior IC software engineer (ML frameworks)
Builds
PyTorch and Triton software stacks for Graphcore hardware
Domain
AI/ML infrastructure, hardware-accelerated computing
Deliverable
production ML models
Required skills
Python development, C++ application development, ML kernel development, compiler development, ML framework development, high-performance computing
Preferred skills
AI/ML knowledge, computing architecture understanding, complex codebase maintenance, high-performance code profiling and optimization
Technologies
PyTorch, Triton, C++, Python
Responsibilities
Designing and implementing new features, developing and maintaining unit/component/integration tests, optimizing software stack performance, managing complex technical tasks with cross-team dependencies, managing code quality and technical debt, providing technical leadership and contributing to team direction, contributing to documentation, conducting code reviews, resolving regressions and defects, coaching and mentoring team members, working with upstream open-source teams
Seniority
Senior, hands-on IC
