Upvote
Downvote
Software Development Manager, Machine Learning Acceleration, Neuron ML Apps
Share Job
- Suggest Revision
Full-time
- AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine
- As the Software Development Manager for the ML Applications - Framework team, you will be responsible for leading a strong team of engineers to help design and deploy ML applications/usecases on various frameworks such as Pytorch, JAX, Tensorflow.
- You will be responsible for the full development life cycle of our integrations and extensions for inference and training support in Pytorch, XLA, Tensorflow and JAX. Develop reliability/scalability features and performance updates in the Neuron ML Frameworks as well as contribute to other popular open Frameworks to enable them make Trainium and Inferentia devices as the first-class citizens for ML Acceleration.
- A successful candidate will have an established background in developing Machine Learning frameworks using Pytorch on XLA devices and corresponding framework technology components such as Torch-XLA, Open-XLA project integrations using PJRT or StableHLO, alongwith familarity of OpenXLA compilers.
- This organization builds the full stack of software, servers and chips to accelerate at the highest scale.
Active Job
Updated TodaySimilar Job
Relevance
Active