Sr. ml kernel performance engineer, aws neuron, annapurna labs
TorontoAmazon Development Centre Canada ULC
...* Experience with LLVM/MLIR backend development for GPUs - * Knowledge of ML frameworks (PyTorch, TensorFlow) and their GPU backends - * Experience with parallel programming and optimization techniques - * Understanding of GPU memory hierarchies and optimization strategies Amazon is an equal opportunity [...]
Category Restaurant, Catering & Hospitality
30+ days ago in AmazonJobs