Machine Learning Performance Engineer, Annapurna Labs
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
Track your applications on mobile The free Whileresume app, on iPhone and Android.
The role
Design, implement, and maintain software solutions to maximize end-to-end ML performance across the full software stack.
Analyze and optimize high-performance ML workloads running on accelerator hardware, focusing on instruction scheduling, memory hierarchy, and parallelism.
Develop and optimize high-performance kernels for core ML operations used by popular frameworks.
Enhance the Neuron SDK and related tooling to improve developer experience and system capabilities.
Collaborate across Compiler, Frameworks, and Hardware teams to diagnose bottlenecks and drive performance improvements.
Work in a startup-like environment, delivering high-impact solutions and influencing technical decisions with measurable impact.
Analyze and optimize high-performance ML workloads running on accelerator hardware, focusing on instruction scheduling, memory hierarchy, and parallelism.
Develop and optimize high-performance kernels for core ML operations used by popular frameworks.
Enhance the Neuron SDK and related tooling to improve developer experience and system capabilities.
Collaborate across Compiler, Frameworks, and Hardware teams to diagnose bottlenecks and drive performance improvements.
Work in a startup-like environment, delivering high-impact solutions and influencing technical decisions with measurable impact.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
or
Already have an account?
Log inYou might also like these jobs
No closely matching jobs yet — here are the most recent ones.