Direct full-stack Software (SW) role, focusing on ML compiler and inference infrastructure.
Design and develop ML compiler stack to bridge AI workloads and low-level Hardware (HW) operators.
Design and develop SW infrastructure to enable high performance inference.
Drive SW/HW codesign, kernel optimization, performance tuning, and debugging.
Minimum qualifications
Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience.
Experience in software development using Python and C++.
Experience with machine learning (ML) hardware accelerators, ML compiler stacks (e.g., JAX, PyTorch, XLA), and kernel development.
Preferred qualifications
PhD degree in Computer Engineering, Computer Science, or a related field.
2 years of experience with full-stack system development for hardware accelerators.
2 years of experience with collaborative, hyper-scalar software development and engineering best practices.
Experience scoping, planning, and executing projects towards team goals in ambiguous environments.
Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience.
Experience in software development using Python and C++.
Experience with machine learning (ML) hardware accelerators, ML compiler stacks (e.g., JAX, PyTorch, XLA), and kernel development.