About this opportunity
Amazon Web Services (AWS) lists this ML & Compiler Engineer for AI Accelerators opportunity in cupertino, California. Review the employer’s description below for duties, qualifications and application requirements.
Job description
Annapurna Labs (U.S.) Inc. is seeking a software engineer to join the AWS Neuron compiler team in Cupertino, CA.
You will build the next generation Neuron compiler, transforming ML models from frameworks like PyTorch, TensorFlow and JAX to run on Inferentia/Trainium in the cloud. You will work with OpenXLA, StableHLO and MLIR, tackle compiler optimization challenges for large language models and vision models, and partner with internal/external customers to deliver high-performance, easy-to-use
#J-18808-Ljbffr
Worksite address
cupertino, CA, 95014, US
Who can apply
Review the original listing for work authorization, qualifications and employer requirements.