Senior / Staff Machine Learning Engineer
Company: ARM
Job location: Chesterton, UK
Posted: 1d ago
Job details
Job Description: Arm's Machine Learning Group is seeking highly motivated and creative Software Engineers to join the Cambridge-based ML Content, Algorithms and Tools team. This Machine Learning Engineer role focuses on advancing the field of AI by optimizing and deploying pioneering models, particularly Large Language Models (LLMs) and Generative AI algorithms. This involves deep analysis of neural networks, optimizing software and hardware, developing innovative solutions, and collaborating with teams to build high-performance AI systems.
Responsibilities:
1. Work with major ML frameworks (PyTorch, TensorFlow, etc.) to port and develop ML networks.
2. Optimize and quantize models for efficient execution on Arm platforms.
3. Ensure multiple Arm products are designed to perform effectively for machine learning.
4. Deeply understand the complex applications you analyze and communicate them in their simplest form to contribute to product designs.
Required Skills and Experience:
1. A background in computer science, software engineering, or other comparable skills.
2. Experience training and debugging neural networks with TensorFlow and PyTorch using Python.
3. Understanding, deploying, and optimizing Large Language Models (LLMs) and Generative AI algorithms.
4. Experience using software development platforms and continuous integration systems.
5. Familiarity with Linux and cloud services.
6. Strong attention to detail to ensure use cases are well understood and critical areas needing improvement are identified.
Nice To Have Skills and Experience:
1. Experience of the inner workings of PyTorch, TensorFlow, Executorch, and TensorFlow Lite.
2. Experience of developing and maintaining CI/testing components to improve automation of model analysis.
3. Good knowledge of Python for working with ML frameworks.
4. Good knowledge of C++ for working with optimized ML libraries.
5. Previous experience of machine learning projects.
6. Experience with deployment optimizations on machine learning models.
In Return:
This role offers a phenomenal opportunity to contribute to the full life cycle of machine learning projects and understand how innovative machine learning is used to solve real-world problems. You will work closely with experts in ML and software and hardware optimization in a truly multi-discipline environment.
Accommodations at Arm:
If you need support or an accommodation to Be Your Brilliant Self during the recruitment process, please email accommodations@arm.com. All accommodation requests will be treated with confidentiality.
Hybrid Working at Arm:
Arm’s approach to hybrid working supports both high performance and personal wellbeing. We empower teams to determine their own hybrid working patterns based on their needs.
Equal Opportunities at Arm:
Arm is an equal opportunity employer, committed to providing an environment of mutual respect where equal opportunities are available to all applicants and colleagues.
#J-18808-Ljbffr