Applied Scientist, AWS Neuron Science team
Core
Enhance software stack for AWS Trainium and Inferentia accelerators by identifying adoption barriers and implementing optimization solutions.
Role type
Applied Scientist (ML Systems & Compilers)
Builds
ML compilers, high-performance kernels, and system robustness tools for AWS ML chips.
Domain
Cloud computing, Machine Learning, Hardware acceleration
Deliverable
production ML models | product features
Required skills
Java, C++, Python, algorithms and data structures, numerical optimization, parallel and distributed computing, high-performance computing
Preferred skills
building machine learning models, developing algorithms for business applications, professional software development
Technologies
Trainium, Inferentia, AWS Neuron, ML frameworks
Responsibilities
Collaborate with engineering teams to implement innovative solutions, engage with academic and research communities, develop ML/RL approaches for kernel optimization, create advanced compiler techniques for ML workloads.
Seniority
Senior, hands-on IC
