Machine Learning Software Engineer
Join an engineering team developing and optimizing software that enables machine learning and AI workloads to run efficiently on next-generation Arm compute platforms. The role spans the ML software stack and may focus on compilers, framework integration, runtimes, drivers, performance analysis, or1
Обязанности
- Develop and optimize C++ and Python software for machine learning workloads
- Work with ML software and hardware engineers to understand model execution and improve performance and efficiency
- Contribute to ML compilation, PyTorch integration, runtimes, drivers, performance analysis, or embedded software projects
- Contribute to technical design and own larger engineering challenges
- Support the development of other engineers
Требования
- Professional experience developing C++ software and complex software systems
- Practical experience with machine learning frameworks or tooling such as PyTorch
- Python experience for software development, ML tooling, testing, or automation
- Experience debugging and performing software performance analysis
- Ability to solve complex software problems
- Experience with source control, code review, testing, and continuous integration
- Collaborative approach to software development
Будет плюсом
- Experience with ML compilers
- Knowledge of PyTorch internals
- Experience with runtimes or drivers
- Embedded software experience
- Performance optimization experience
- Hardware acceleration knowledge
- Experience providing technical leadership or mentoring
Условия и преимущества
- Continuous learning and career development
- Collaborative global engineering environment
- Comprehensive employee benefits
- Hybrid working arrangements determined by the team
- Recruitment adjustments and accommodations available