Senior Machine Learning Engineer, AWS Neuron Apps
Amazon Web Services (AWS)Job Overview
Amazon Web Services (AWS) is seeking a Senior Machine Learning Engineer to join the AWS Neuron Apps team, focusing on ML Inference Applications. This role involves the development, enablement, and performance tuning of a wide variety of ML model families, including large language models like Llama2, GPT2, GPT3, stable diffusion, and Vision Transformers.
Responsibilities
- Lead the efforts in building distributed inference support into Pytorch and TensorFlow using XLA and the Neuron compiler and runtime stacks.
- Tune models to ensure highest performance on AWS Trainium and Inferentia silicon and TRn1, Inf1 servers.
- Engage in software development using C++/Python and apply ML knowledge to optimize inference performance for both latency and throughput.
Day-to-Day Activities
- Design and code solutions to drive efficiencies in software architecture.
- Create metrics, implement automation, and resolve software defects.
- Participate in design discussions, code reviews, and communicate with internal and external stakeholders.
- Work cross-functionally to help drive business decisions with your technical input.
About the Team
The ML Apps team works closely with compiler engineers and runtime engineers to create, build, and tune distributed inference solutions. The team environment supports new members through mentorship and knowledge-sharing, fostering an inclusive and collaborative work culture.
Career Growth and Development
AWS values diverse experiences and is committed to building a team that represents a variety of backgrounds, perspectives, and skills. The role offers significant opportunities for career growth, aiming to help team members develop into well-rounded professionals capable of taking on complex tasks in the future.
Work/Life Balance
AWS promotes work-life harmony by offering flexible work arrangements, ensuring that success at work does not come at the expense of your personal life.
Inclusive Team Culture
Our team embraces diversity and inclusion, actively fostering a workplace where everyone feels valued and inspired. We support various employee-led affinity groups and host ongoing learning experiences to promote an inclusive environment.
Benefits Extracted with AI
- Health insurance
- Paid vacation
- Flexible work hours
- Remote work options
Similar jobs
Last update: 23 minutes ago
Applied Scientist, Artificial General Intelligence
Join AWS as an Applied Scientist in Artificial General Intelligence, driving AI innovation in cloud computing.
Senior Applied Scientist, AWS Marketing AI/ML
Join AWS as a Senior Applied Scientist in Marketing AI/ML, leading personalization and targeting initiatives.
Senior Distributed Systems Engineer
Join webAI as a Senior Distributed Systems Engineer to design and maintain scalable systems using Python, Kubernetes, and more.
Machine Learning Engineer with AI/ML Experience
Join us as a Machine Learning Engineer to develop AI/ML models and applications. Work remotely with top-tier companies.
Senior Machine Learning Engineer
Join Amazon as a Senior Machine Learning Engineer to build scalable AI/ML infrastructure and MLOps platforms.
Senior Machine Learning Engineer
Join as a Senior Machine Learning Engineer to design and deploy advanced ML solutions using Python, Spark, and cloud platforms. Remote work opportunity.
AI Framework Engineer
Join webAI as an AI Framework Engineer to develop innovative AI frameworks for distributed computing environments.
Senior Applied Scientist
Join Amazon as a Senior Applied Scientist to lead machine learning projects, optimize ad performance, and drive innovation.
Senior Python Engineer (Cloud Platform)
Join Bonfy.AI as a Senior Python Engineer to build and maintain a cloud-based SaaS platform using Python and AWS.
Senior Applied Scientist
Join Amazon as a Senior Applied Scientist to lead machine learning projects, optimize models, and drive innovation in advertising.
Senior Worldwide Specialist, GenAI Model Training & Inference
Join AWS as a Senior Specialist in GenAI Model Training & Inference, driving customer adoption and scaling workloads.
Frontend Engineer II
Join AWS as a Frontend Engineer II to build web applications using Angular, CSS, and JavaScript frameworks.
Senior Software Engineer, Machine Learning
Join as a Senior Software Engineer in Machine Learning, working remotely to build ML-driven products for user engagement.
Senior Applied Scientist, Sponsored Products
Join Amazon as a Senior Applied Scientist in New York, focusing on AI, ML, and data science for Sponsored Products.
Senior Machine Learning Engineer
Join Nike as a Senior Machine Learning Engineer to develop and optimize ML algorithms for innovative applications.
Applied Scientist, Artificial General Intelligence
Join Amazon's AGI team as an Applied Scientist to develop cutting-edge AI technology in Computer Vision and NLP.
Senior Cloud Engineer
Join as a Senior Cloud Engineer to architect and deploy cloud solutions using Azure, AWS, and GCP. Lead innovation in cloud technology.
Senior Software Development Engineer, Customer Trust - Core Services
Join Amazon as a Senior Software Development Engineer to build scalable AI systems for customer trust.
Senior Machine Learning Engineer
Join NIQ as a Senior ML Engineer to develop and implement AI models using Python, PyTorch, and Azure in a hybrid work environment.
Senior Machine Learning Engineer
Join Atypon as a Senior ML Engineer to develop AI solutions in NLP, deep learning, and MLOps. Remote position in Athens.
Machine Learning Engineer for Predictive Metrics, Alexa Conversational Assistant Services
Join Amazon as a Machine Learning Engineer to develop predictive metrics for Alexa using Generative AI and AWS Inferentia2.
Senior Software Engineer, Platform
Join Augment AI as a Senior Software Engineer to build AI-driven platforms using AWS, Ruby, and Python. Enjoy great benefits and stock options.
Senior Fullstack Software Engineer, GenAI Horizontal Task Tooling
Join Scale AI as a Senior Fullstack Software Engineer to build web-based applications for AI data annotation.
Machine Learning Engineer - University Graduate 2025
Join Adobe as a Machine Learning Engineer to develop AI technologies for Photoshop and Digital Imaging.