The AWS Neuron Science Team is looking to enhance their software stack and accelerate customer adoption of Trainium and Inferentia accelerators by identifying key adoption barriers and optimization opportunities.
Requirements
- Experience programming in Java, C++, Python or related language
- Experience with neural deep learning methods and machine learning
- Experience with modeling tools such as R, scikit-learn, Spark MLLib, MxNet, Tensorflow, numpy, scipy etc.
Responsibilities
- AI for Systems: Developing and applying ML/RL approaches for kernel/code generation and optimization
- Machine Learning Compiler: Creating advanced compiler techniques for ML workloads
- System Robustness: Building tools for accuracy and reliability validation
- Efficient Kernel Development: Designing high-performance kernels optimized for our ML accelerator architectures
Other
- PhD, or Master's degree and 6+ years of applied research experience
- 3+ years of building machine learning models for business application experience
- Work safely and cooperatively with other employees, supervisors, and staff
- Adhere to standards of excellence despite stressful conditions
- Communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service
