Amazon's AWS Neuron team in Cupertino, CA seeks a senior software engineer to advance the next generation of a compiler transforming PyTorch, TensorFlow and JAX models into efficient deployments on Inferentia and Trainium servers.
As part of a startup‑like environment within AWS, you will design and implement optimization passes, collaborate across chip, runtime, and ML teams, and contribute to open‑source communities while mentoring others. Expect rigorous code reviews and robust testing.
#J-18808-LjbffrSenior ML Compiler Engineer – Build Next‑Gen ML on AI Accelerators in cupertino at Unknown Company
This position is listed as full time and onsite.