> Markdown version of [/jobs/ext/1247062-sr-software-engineer-ai-ml-aws-neuron-apps](https://www.wearedevelopers.com/jobs/ext/1247062-sr-software-engineer-ai-ml-aws-neuron-apps). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md). --- # Sr. Software Engineer- AI/ML, AWS Neuron Apps - **Company:** Amazon.com, Inc. - **Location:** Seattle, WA, United States - **Experience:** Expert - **Salary:** $83,200.0 - $104,000.0 - **Contract:** Permanent contract - **Skills:** Training Data, Artificial Intelligence, Amazon Web Services, C++ (Programming Language), COBOL (Programming Language), Program Optimization, Code Review, Nvidia CUDA, Computer Programming, Software Debugging, Distributed Systems, Python (Programming Language), Machine Learning, Open Source Technology, Performance Tuning, Tensorflow, Software Engineering, Systems Architecture, AI Infrastructure, Pytorch, Large Language Models, Parallel Computation, Information Technology, Build Process, Hardware Acceleration, Machine Learning Operations, Front End Software Development, Software Coding, GPT, Software Version Control - **Published:** July 12, 2026 - **Apply:** https://www.careerjet.com/job/usc2f68a1c476620fa448b30f70a1107ed/eaa ## About the Role 5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience - 5+ years of programming experience using Python or C++ and PyTorch. - Experience with AI acceleration via quantization, parallelism, model compression, batching, KV caching, vllm serving - Experience with accuracy debugging & tooling, performance benchmarking of AI accelerators - Fundamentals of Machine learning and deep learning models, their architecture, training and inference lifecycles along with work experience on optimizations for improving the model execution. Preferred Qualifications - Master's degree in computer science or equivalent - Master's degree in machine learning or equivalent - Experience with accuracy debugging & tooling, performance benchmarking of AI accelerators - Experience in developing CUDA kernels, HPC and inference optimization, tensors operations ## Description Join the elite team behind AWS Neuron-the software stack powering AWS's next-generation AI accelerators Inferentia and Trainium. As a Senior Software Engineer in our Machine Learning Applications team, you'll be at the forefront of deploying and optimizing some of the world's most sophisticated AI models at unprecedented scale. What You'll Impact: * Pioneer distributed inference solutions for industry-leading LLMs such as GPT, Llama, Qwen * Optimize breakthrough language and vision generative AI models * Collaborate directly with silicon architects and compiler teams to push the boundaries of AI acceleration * Drive performance benchmarking and tuning that directly impacts millions of inference calls globally, You will drive the Evolution of Distributed AI at AWS Neuron As a Technical Leader at the forefront of AWS's AI Accelerator, you'll architect the bridge between ML frameworks including PyTorch, JAX and AI hardware. This isn't just about just optimization-it's about revolutionizing how AI models run at scale. Technical Impact You'll Drive: * Spearhead distributed inference architecture for PyTorch and JAX using XLA * Engineer breakthrough performance optimizations for AWS Trainium and Inferentia * Develop ML tools to enhance LLM accuracy and efficiency * Transform complex tensor operations into highly optimized hardware implementations * Pioneer benchmarking methodologies that shape next-gen AI accelerator design What Makes This Role Unique: * Direct influence on AWS's AI infrastructure used by thousands of ML applications * Full-stack optimization from high-level frameworks to hardware-specific primitives * Creation of tools and frameworks that define industry standards for ML deployment * Collaboration with both open-source ML communities and hardware architecture teams Your Technical Arsenal Should Include: * Deep expertise in Python and ML framework internals * Strong understanding of distributed systems and ML optimization * Passion for performance tuning and system architecture A day in the life Work/Life Balance Our team puts a high value on work-life balance. It isn't about how many hours you spend at home or at work; it's about the flow you establish that brings energy to both parts of your life. We believe striking the right balance between your personal and professional life is critical to life-long happiness and fulfillment. We offer flexibility in working hours and encourage you to find your own balance between your work and personal lives. Mentorship & Career Growth Our team is dedicated to supporting new members. We have a broad mix of experience levels and tenures, and we're building an environment that celebrates knowledge sharing and mentorship. We care about your career growth and strive to assign projects based on what will help each team member develop into a better-rounded professional and enable them to take on more complex tasks in the future. About the team At AWS Neuron, we're revolutionizing how the world's most sophisticated AI models run at scale through Amazon's next-generation AI accelerators. Operating at the unique intersection of ML frameworks and custom silicon, our team drives innovation from silicon architecture to production software deployment. We pioneer distributed inference solutions for PyTorch and JAX using XLA, optimize industry-leading LLMs like GPT and Llama, and collaborate directly with silicon architects to influence the future of AI hardware. Our systems handle millions of inference calls daily, while our optimizations directly impact thousands of AWS customers running critical AI workloads. We're focused on pushing the boundaries of large language model optimization, distributed inference architecture, and hardware-specific performance tuning. Our deep technical experts transform complex ML challenges into elegant, scalable solutions that define how AI workloads run in production., Join a dynamic team as a Frontend Software Engineer, where you will play a crucial role in developing exceptional user interfaces for high-impact customer projects. Your expertise … + 5 hours ago, Join a dynamic team as a Software Engineer specializing in COBOL, where you will play a crucial role in enhancing high-quality training data for advanced AI systems. This is an exc… + 8 days ago + ## Related Videos - [Coffee with Developers - Stephen Jones - NVIDIA](https://www.wearedevelopers.com/videos/1303-coffee-with-developers-stephen-jones-nvidia) - [Machine learning in the browser with TensorFlowjs](https://www.wearedevelopers.com/videos/155-machine-learning-in-the-browser-with-tensorflowjs) - [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding) - [ Evaluating AI models for code comprehension](https://www.wearedevelopers.com/videos/1462-evaluating-ai-models-for-code-comprehension) - [Alibaba Big Data and Machine Learning Technology](https://www.wearedevelopers.com/videos/37-alibaba-big-data-and-machine-learning-technology) - [Architecting the Future: Leveraging AI, Cloud, and Data for Business Success](https://www.wearedevelopers.com/videos/1096-architecting-the-future-leveraging-ai-cloud-and-data-for-business-success) ## Related Articles - [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development) - [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer) - [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers) - [Got AI ideas but no money? Here are 10 free ways to level up your AI skills with Google Cloud](https://www.wearedevelopers.com/magazine/600-got-ai-ideas-but-no-money-here-are-10-free-ways-to-level-up-your-ai-skills-with-google-cloud) - [Best US AI Conferences for CTOs in 2026: Build vs. Buy, Vendor Evaluation, and Peer Intelligence](https://www.wearedevelopers.com/magazine/736-best-us-ai-conferences-for-ctos-in-2026-build-vs-buy-vendor-evaluation-and-peer-intelligence) - [Coffee with Developers - Maria Apazoglou - Making AI understandable for all in production](https://www.wearedevelopers.com/magazine/475-coffee-with-developers-maria-apazoglou-making-ai-understandable-for-all-in-production)