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-LjbffrML & Compiler Engineer for AI Accelerators in cupertino at Unknown Company
This position is listed as full time and onsite.