Senior Applied Scientist, AWS Neuron Science - Core Algorithm
Core
Design and optimize hardware-aware machine learning algorithms for AWS Trainium and Inferentia accelerators, focusing on low-precision training, efficient distributed systems, and foundation model pre-training.
Role type
Senior Applied Scientist (Hardware-Aware ML)
Builds
Production ML models and algorithms for AWS custom silicon (Trainium/Inferentia)
Domain
Cloud computing, Machine Learning, Hardware Co-design (via careerplan.io/jobs/10570541-senior-applied-scientist-aws-neuron-science-core-algorithm-at-amazon)
Deliverable
production ML models
Required skills
Deep learning, Machine learning, Large-scale distributed systems, Low-precision arithmetic (MXFP8/MXFP4), PyTorch, JAX, C++, Python
Preferred skills
R, scikit-learn, Spark MLLib, MxNet, Tensorflow, numpy, scipy, Hadoop
Responsibilities
Own scientific problems end-to-end from research to production, develop production-quality code for large-scale training systems, partner with engineering and hardware teams, mentor scientists and interns, publish research at top venues
Seniority
Senior, hands-on IC