Software Engineer, Accelerators
Core
Building low-level software and high-performance kernels to enable large-scale AI training and inference on advanced supercomputing platforms and non-traditional hardware.
Role type
Senior IC systems engineer specializing in AI accelerators and distributed systems
Builds
Production AI software stack, high-performance kernels, and distributed system optimizations for frontier AI workloads
Domain
AI Infrastructure / High-Performance Computing / Hardware-Software Co-design
Deliverable
production ML models
Required skills
AI infrastructure development, accelerator platform bring-up, kernel development, distributed system scaling, performance modeling, system debugging, software stack adaptation, PyTorch optimization
Preferred skills
Data center-scale AI hardware experience, custom silicon expertise, mobile accelerator exposure
Technologies
PyTorch, TPUs, custom silicon, distributed systems frameworks
Responsibilities
Prototype and enable AI software stack on new accelerator platforms, optimize large-scale model performance for diverse hardware, develop kernels and sharding mechanisms, collaborate on model code and runtime optimizations, perform system-level performance modeling and bottleneck debugging, work with hardware teams to adapt software to new architectures
Seniority
Senior, hands-on IC
