Human role

Machine Learning Performance Engineer, Annapurna Labs

Amazoncom

Tel Aviv-Yafo, Tel Aviv, ISR aijobs Today
Apply now

The role

Job description

Analyze high performance ML workloads on accelerator hardware; Build and maintain innovative software solutions; Create performance metrics; Develop high performance kernels for ML operations; Enhance Neuron SDK; Implement automation to improve performance; Implement compiler enhancements; Improve instruction scheduling; Improve memory management; Optimize ML system performance across software stack; Optimize parallelism; Participate in design discussions and code reviews; Perform root cause ana

View full posting

Index terms

Skills & keywords

C++Co-designCompiler optimizationHardware-Software Co-designHardware/softwareInstruction schedulingJAXKernel optimizationLLMMachine LearningMemory ManagementParallelism

Get roles like this in your inbox

New agentic AI jobs, curated every Thursday. No spam.

Explore more

All General AI jobs
AmazoncomMachine Learning Performance Engineer, Annapurna Labs
Apply