Upvote
Downvote
Sr. Software Engineer - AI / ML, AWS Neuron Distributed Training
Share Job
- Suggest Revision
- This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron.
- This role is responsible for development, enablement and performance tuning of a wide variety of ML model families, including massive scale large language models like GPT2, GPT3 and beyond, as well as stable diffusion, Vision Transformers and many more.
- This role will help lead the efforts building distributed training and inference support into Pytorch, Tensorflow, Jax using XLA and the Neuron compiler and runtime stacks.
- We have ten employee-led affinity groups, reaching 40,000 employees in over 190 chapters globally.
- It isn’t about how many hours you spend at home or at work; it’s about the flow you establish that brings energy to both parts of your life.
Active Job
Updated 8 days agoSimilar Job
Relevance
Active