The AWS Neuron Science Team is looking to enhance its software stack to accelerate customer adoption of Trainium and Inferentia accelerators.
Requirements
- PhD, or a Master's degree and experience building machine learning models or developing algorithms for business application
- Experience in patents or publications at top-tier peer-reviewed conferences or journals
- Experience in any of the following areas: algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, high-performance computing
- Experience programming in Java, C++, Python or related language
- Experience using Unix/Linux
- Experience in professional software development
Responsibilities
- Developing and applying ML/RL approaches for kernel/code generation and optimization
- Creating advanced compiler techniques for ML workloads
- Building tools for accuracy and reliability validation
- Designing high-performance kernels optimized for our ML accelerator architectures
- Work directly with external and internal customers to identify key adoption barriers and optimization opportunities
- Collaborate closely with our engineering teams to implement innovative solutions
- Engage with academic and research communities to advance state-of-the-art ML systems
Other
- 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
- follow all federal, state, and local laws and Company policies
- exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation
