Amazon Development Centre Canada ULC
Amazon Development Centre Canada ULC

ML Kernel Performance Engineer, AWS Neuron, Annapurna Labs

Salary

Apply with your own terms

Job status
Permanent
Location
Amazon Development Centre Canada ULC
Toronto, ON
Workplace
On site
Job description

Join the Annapurna Labs team at Amazon Web Services (AWS) as an ML Kernel Performance Engineer. We build AWS Neuron, a software development kit to accelerate deep learning and GenAI workloads. Our Acceleration Kernel Library team maximizes performance for AWS's custom ML accelerators, working at the hardware-software boundary to craft high-performance kernels that ensure every FLOP counts. Our comprehensive toolkit integrates with popular ML frameworks like PyTorch, optimizing current performance and contributing to future architecture designs. Collaborate with customers and shape the future of AI acceleration technology.

Share this job

? Join our team and work at the forefront of ML acceleration! As an ML Kernel Performance Engineer, you'll optimize AWS's custom ML accelerators at the hardware-software boundary. Dive deep into designing high-performance kernels and innovate with us at AWS Annapurna Labs. Collaborate, learn, and grow in an inclusive and diverse environment, with mentorship and a focus on work-life balance. Become part of a team shaping the future of AI acceleration technology! ?

Apply for this position

Job offer are open to all, without discrimination and in gender equality.

Throughout the website, gender-specific terms may be used in order to ease the text flow.

We use cookies to ensure that we give you the best experience on our website.
By continuing to use our site, you accept our cookies policy terms
Select cookies to accept

Please uncheck the cookies you don't like.

Consent cookies

These cookies are only used to remember your consent settings and cannot be unchecked. Unchecking this option would result in your consent settings not being applied to the website.

Session cookies

These cookies are used to identify whether you are logged in to the website and to identify your open session. They are only created when you log in or register on the website and therefore cannot be unchecked. By logging in or registering on the website, you agree to our cookie policy.

Performance

These cookies allow us to analyze navigation on our site to improve its operation.

Targeted advertising

These cookies help us limit the number of times you see an advertisement, personalize our offers and services based on your interests, measure the effectiveness of an advertising campaign, etc. These can be shared with our partners.