Mediabistro logo
job logo

Amazon is hiring: Senior Machine Learning Engineer, Model Customization, Generat

Amazon, San Francisco, CA, United States


Job ID: 3141363 | Amazon Web Services, Inc.

The Generative AI Innovation Center at AWS empowers customers to harness state of the art AI technologies for transformative business opportunities. Our multidisciplinary team of strategists, scientists, engineers, and architects collaborates with customers across industries to fine‑tune and deploy customized generative AI applications at scale. Additionally, we work closely with foundational model providers to optimize AI models for Amazon Silicon, enhancing performance and efficiency.

As an SDE on our team, you will drive the development of custom Large Language Models (LLMs) across languages, domains, and modalities. You will be responsible for fine‑tuning state‑of‑the‑art LLMs for diverse use cases while optimizing models for high‑performance deployment on AWS’s custom AI accelerators. This role offers an opportunity to innovate at the forefront of AI, tackling end‑to‑end LLM training pipelines at massive scale and delivering next‑generation AI solutions for top AWS clients.

Key job responsibilities

Large‑Scale Training Pipelines: Design and implement distributed training pipelines for LLMs using tools such as Fully Sharded Data Parallel (FSDP) and DeepSpeed, ensuring scalability and efficiency

LLM Customization & Fine‑Tuning: Adapt LLMs for new languages, domains, and vision applications through continued pre‑training, fine‑tuning, and Reinforcement Learning with Human Feedback (RLHF)

Model Optimization on AWS Silicon: Optimize AI models for deployment on AWS Inferentia and Trainium, leveraging the AWS Neuron SDK and developing custom kernels for enhanced performance

Customer Collaboration: Interact with enterprise customers and foundational model providers to understand their business and technical challenges, co‑developing tailored generative AI solutions

About the team
ABOUT AWS: Diverse Experiences. Amazon values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying. Why AWS: Amazon Web Services (AWS) is the world’s most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and never stopped innovating — that’s why customers from the most successful startups to Global 500 companies trust our robust suite of products and services to power their businesses. Work/Life Balance: We value work‑life harmony. Achieving success at work should never come at the expense of sacrifices at home, which is why we strive for flexibility as part of our working culture. When we feel supported in the workplace and at home, there’s nothing we can’t achieve in the cloud. Inclusive Team Culture: AWS values curiosity and connection. Our employee‑led and company‑sponsored affinity groups promote inclusion and empower our people to take pride in what makes us unique. Our inclusion events foster stronger, more collaborative teams. Mentorship and Career Growth: We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge‑sharing, mentorship and other career‑advancing resources here to help you develop into a better‑rounded professional. AWS Global Services: AWS Global Services includes experts from across AWS who help our customers design, build, operate, and secure their cloud environments. Customers innovate with AWS Professional Services, upskill with AWS Training and Certification, optimize with AWS Support and Managed Services, and meet objectives with AWS Security Assurance Services. Our expertise and emerging technologies include AWS Partners, AWS Sovereign Cloud, AWS International Product, and the Generative AI Innovation Center. You’ll join a diverse team of technical experts in dozens of countries who help customers achieve more with the AWS cloud.

Basic Qualifications

5+ years of non‑internship professional software development experience

5+ years of programming with at least one software programming language experience

5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience

Experience as a mentor, tech lead or leading an engineering team

Hands‑on experience with deep learning and machine learning methods (e.g., for training, fine tuning, and inference)

Experience with design, development, and optimization of generative AI solutions, algorithms, or technologies

Preferred Qualifications

5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience

Bachelor’s degree in computer science or equivalent

2+ years of building machine learning models or developing algorithms for business application experience

Hands‑on experience with at least one ML library or framework

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Los Angeles County applicants: Job duties for this position include: 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; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to 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. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

Our compensation reflects the cost of labor across several US geographic markets. The base pay for this position ranges from $151,300/year in our lowest geographic market up to $261,500/year in our highest geographic market. Pay is based on a number of factors including market location and may vary depending on job‑related knowledge, skills, and experience. Amazon is a total compensation company. Dependent on the position offered, equity, sign‑on payments, and other forms of compensation may be provided as part of a total compensation package, in addition to a full range of medical, financial, and/or other benefits. For more information, please visit https://www.aboutamazon.com/workplace/employee-benefits. This position will remain posted until filled. Applicants should apply via our internal or external career site.

Posted: August 24, 2025 (Updated about 3 hours ago)

#J-18808-Ljbffr

In Summary: The Generative AI Innovation Center at Amazon Web Services empowers customers to harness state of the art AI technologies for transformative business opportunities . The role offers an opportunity to innovate at the forefront of AI, tackling end‑to‑end LLM training pipelines at massive scale and delivering next‑generation AI solutions for top clients .

En Español: ID de trabajo: 3141363  Amazon Web Services, Inc. El Centro de Innovación de IA Generativa en AWS empodera a los clientes para aprovechar las tecnologías AI más avanzadas para oportunidades empresariales transformadoras. Nuestro equipo multidisciplinario de estrategas, científicos, ingenieros y arquitectos colabora con clientes en todas las industrias para ajustar y implementar aplicaciones generativas de IA personalizadas a escala. Además, trabajamos estrechamente con proveedores de modelos fundamentales para optimizar modelos de IA para Amazon Silicon, mejorando el rendimiento y la eficiencia. Como SDE en nuestro equipo, impulsará el desarrollo de Modelos Large Language (LLM) personalizados en diferentes idiomas, dominios y modalidades. Usted será responsable del ajuste fino del estado de los LL-artMs para casos de uso al tiempo que optimiza varios modelos de implementación de alto desempeño en aceleradores personalizados de AI de AWS. Este papel ofrece la oportunidad de innovar a la vanguardia de la IA, abordando pipelines de capacitación LLM de extremo a extremo en escala masiva y ofreciendo soluciones AI de próxima generación para los principales clientes de AWS. Responsabilidades clave del trabajo Pipelines: Diseño e implementación de tuberías distribuidas de formación para LLMs utilizando herramientas como Full Sharded Data Parallel (FSDP) y DeepSpeed, asegurando la escalabilidad y eficiencia La personalización y el ajuste fino de LLM: Adapta las LLMs para nuevos idiomas, dominios y aplicaciones mediante una continua pre-entrenamiento, ajustación fina y refuerzo de aprendizaje con retroalimentación humana (RLHF). Optimización del modelo de trabajo: Optimizar modelos de AWS para mejorar su carrera en AWS Inferentia y Trainia, aprovechando los servicios de AWS Neels ABOUT para desarrollar un mayor rendimiento completo: Por qué no se han adoptado nuevas experiencias tecnológicas desde el punto de vista humano? Hemos sido pioneros en la computación en nube y nunca hemos dejado de innovar. Es por eso que los clientes, desde las startups más exitosas hasta las compañías Global 500, confían en nuestro sólido conjunto de productos y servicios para impulsar sus negocios. Equilibrio trabajo/vida: Valoramos la armonía entre vida laboral y personal. Lograr el éxito en el trabajo nunca debe venir a expensas de sacrificios en casa, por lo que nos esforzamos por ser flexibles como parte de nuestra cultura de trabajo. Cuando nos sentimos apoyados en el lugar de trabajo y en el hogar, no hay nada que no podamos lograr en la nube. Cultura del Equipo Inclusivo: Curiedad y conexión AWS. Nuestros valores líderes y grupos de afinidad patrocinados por empresas fomentan la inclusión y empoderan a nuestras personas a sentirse orgullosos de lo que hace que somos únicos. Nuestra experiencia y tecnologías emergentes incluyen AWS Partners, AWS Sovereign Cloud, Producto Internacional de AWS y el Centro Generativo de Innovación AI. Se unirá a un equipo diverso de expertos técnicos en docenas de países que ayudan a los clientes a lograr más con la nube AWS. Qualificaciones básicas 5+ años de experiencia profesional en el desarrollo de software sin pasantía Experiencia en programación con al menos un lenguaje de programación de software 5+ año de experiencia en diseño o arquitectura líder (patrones de diseño, fiabilidad y escalado) de sistemas nuevos y existentes Experiencia como mentor, jefe técnico o dirigente de un equipo de ingeniería Experiencia práctica con métodos de aprendizaje profundo y aprendizaje automático (por ejemplo, para capacitación, ajuste fino e inferencia). El historial penal puede tener una relación directa, adversa y negativa con algunos de los deberes laborales materiales de este puesto. Estos incluyen las obligaciones y responsabilidades enumeradas anteriormente, así como la capacidad para adherirse a las políticas de la empresa, ejercer un buen juicio, manejar eficazmente el estrés y trabajar de manera segura y respetuosa con otros, mostrar fiabilidad y profesionalidad, y salvaguardar las operaciones comerciales y la reputación de la Compañía. De acuerdo con la Ordenanza del Condado de Los Ángeles Fair Chance, consideraremos solicitantes calificados de empleo con registros de arresto y condena. Según la Ordeanza de San Francisco Fair Chance , consideraremos candidatos cualificados en empleo con archivos de arrestos y condenas. Nuestra cultura inclusiva empodera a los amazónicos a ofrecer los mejores resultados para nuestros clientes. Si usted tiene discapacidad y necesita alojamiento o lugar de trabajo, por favor póngase en contacto durante el proceso de solicitud de contratación, incluyendo los costos más bajos en nuestro mercado laboral. El salario se basa en una serie de factores, incluida la ubicación del mercado y puede variar dependiendo de los conocimientos, habilidades y experiencia relacionados con el trabajo. Amazon es una compañía de compensación total. Dependiendo de la posición ofrecida, las acciones, pagos por inscripción y otras formas de indemnización pueden ser proporcionadas como parte de un paquete de compensaciones totales, además de una gama completa de beneficios médicos, financieros y / u otros. Para obtener más información, visite https://www.aboutamazon.com/workplace/employee-benefits. Este puesto permanecerá publicado hasta que sea llenado. Los solicitantes deben solicitar a través de nuestro sitio web interno o externo para carreras. Publicado: 24 de agosto de 2025 (actualizado hace aproximadamente 3 horas) #J-18808-Ljbffr