Senior AI/ML Software Engineer - Distributed Training

Amazon; Milano; In loco; EUR 130.000 - 226.000; Tempo pieno

Descrizione del lavoro

A leading technology company in Milan is seeking a Sr. Software Engineer specializing in AI/ML to develop performance optimization solutions for AWS. Candidates should have extensive software development experience and a strong background in machine learning. This role involves collaboration with technical teams to enhance distributed training capabilities and ensure high performance on AWS hardware. Competitive salary and benefits included.

Competenze

• 5+ years of non-internship professional software development experience.
• 5+ years of programming with at least one software programming language experience.
• 5+ years leading design or architecture of new and existing systems experience.
• 5+ years of full software development life cycle experience.
• Experience as a mentor or tech lead.

Mansioni

• Build distributed training and inference support into PyTorch, TensorFlow, Jax.
• Tune models for highest performance on AWS Trainium and Inferentia.
• Collaborate with chip architects and compiler engineers.

Conoscenze

Software development

Machine Learning

Python

Mentoring

Design patterns

Formazione

Bachelor's degree in computer science or equivalent

Descrizione del lavoro

A leading technology company in Milan is seeking a Sr. Software Engineer specializing in AI/ML to develop performance optimization solutions for AWS. Candidates should have extensive software development experience and a strong background in machine learning. This role involves collaboration with technical teams to enhance distributed training capabilities and ensure high performance on AWS hardware. Competitive salary and benefits included.