Neuron Collectives Software Engineer, Trainium Collectives

Job Description

Develop and optimize collective communication algorithms and low-level device drivers to scale AI compute across data centers. Collaborate with hardware teams to co-optimize software and Trainium silicon for maximum performance in LLM training.

Requirements: Requires at least 3 years of professional software development experience and 2 years of experience in system design or architecture. A bachelor's degree in computer science or equivalent is preferred.

Key Skills: C++, C, Collective Communication Patterns, Device Driver Development, Distributed Systems, AI Training Hardware, Algorithm Optimization, System Architecture, Performance Tuning, Software Development Life Cycle, Code Review, Source Control Management, Build Processes, Testing, Operations, Low Level Programming

Benefits: Sign-on payments, Restricted stock units (RSUs), Health insurance, Medical insurance, Dental insurance, Vision insurance, Prescription insurance, Basic Life & AD&D insurance, Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage, 401(k) matching, Paid time off, Parental leave