Neuron Collectives Software Engineer, Trainium Collectives

What you’d do

Develop and optimize C/C++ collective communication algorithms and low-level device drivers for Trainium AI training hardware. Work with hardware teams to improve training performance and analyze compute and bandwidth bottlenecks. Full-time. Found on the employer's careers page on Sep 30, 2026. Apply through the link.