ChatGPT Jobs ยท 5 days ago
Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training
Annapurna Labs is a company that designs silicon and software to accelerate innovation, and they are seeking a Senior Machine Learning Engineer for their Distributed Training team. The role involves developing, enabling, and performance tuning machine learning model families, particularly focusing on large-scale models and distributed training solutions using AWS custom silicon.
Computer Software
Responsibilities
You will lead efforts to build distributed training support into PyTorch, the Neuron compiler, and runtime stacks
You will enable distribute training strategies as well as use them to optimize models to achieve peak performance and maximize efficiency on AWS custom silicon, including Trainium servers
Qualification
Required
Bachelor's degree in computer science or equivalent
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
5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
Experience as a mentor, tech lead or leading an engineering team
Experience in machine learning, large scale training with LLMs and expertise in Pytorch
Preferred
Master's degree in computer science or equivalent
Experience in computer architecture
Previous software engineering expertise with Pytorch/Jax/Tensorflow, Distributed libraries and Frameworks, End-to-end Model Training
Company
ChatGPT Jobs
We find the best job offers for experts in ChatGPT and related technologies.
Funding
Current Stage
Early StageCompany data provided by crunchbase