A
Edge AI ML Kernel Performance Engineer
Amazon Development Centre Canada ULC
Opportunity Description
Drive innovation at Amazon Devices as an ML Kernel Performance Engineer specializing in edge AI. Focus on high-performance CUDA and Triton kernels to optimize GPU performance and model efficiency.
In this pivotal role, you will enhance model training and inference by developing cutting-edge kernel optimizations. Your work will contribute directly to advanced neural network compression platforms and gain significant throughput improvement in deployments. Collaboration with scientists and hardware engineers is key to achieving impactful results in this engineering position.
Key Responsibilities:
• Implement high-performance kernels for quantization tasks
• Optimize kernel processes to minimize training times
• Create a profiling and testing harness for kernels
• Ensure the quality and maintenance of the kernels library
• Collaborate on ML solutions for cloud and edge applications
Requirements:
• Minimum 3 years in...
In this pivotal role, you will enhance model training and inference by developing cutting-edge kernel optimizations. Your work will contribute directly to advanced neural network compression platforms and gain significant throughput improvement in deployments. Collaboration with scientists and hardware engineers is key to achieving impactful results in this engineering position.
Key Responsibilities:
• Implement high-performance kernels for quantization tasks
• Optimize kernel processes to minimize training times
• Create a profiling and testing harness for kernels
• Ensure the quality and maintenance of the kernels library
• Collaborate on ML solutions for cloud and edge applications
Requirements:
• Minimum 3 years in...
Interested in this opportunity? Apply now through Expertini.
Apply for this Position