Staff Machine Learning Engineer

ARM-posted 4 months ago
$191,100 - $258,500/Yr
Mid Level
Hybrid • Austin, TX
Resume
Match Score
Upload and Match ResumeTrack Jobs with Teal

The mission of Central Technology - Machine Learning (CT-ML) is to create technology to enable all ML Compute on ARM. We believe that advancing innovative machine learning (ML) technology requires a collaborative approach across hardware, software, and algorithms. The CT-ML Algorithms team acts as the Center of Excellence for ML content analysis, algorithm development, and tools, driving this unified effort forward.

  • Develop in depth understanding of ML workload, develop algorithms and optimization techniques to drive PPA (Performance, Power, Area) on current and future Arm platforms.
  • Implementation of performance critical machines specific kernel or ARM assembly code on dedicated hardware accelerator and CPU to drive algorithm and architecture exploration.
  • Collaborate with ML Algorithm development team to explore performance critical analysis.
  • Develop internal tooling capabilities to support algorithmic and architecture exploration.
  • 3+ years experience in developing performance critical kernel on dedicated accelerator, GPU or CPU.
  • Graduate students in Computer Engineering, Electrical Engineering, Computer Science or other related technical fields.
  • Deep knowledge in machine learning, deep learning, and neural network design, optimization, and compression techniques.
  • Overall high-level knowledge of computer architecture, systems, and HW-SW co-design.
  • Ability to develop and work with large software systems in programming languages like Python.
  • Knowledge of cutting-edge deep learning libraries such as Tensorflow, and Pytorch.
  • Willing to learn and train large deep learning models on GPU-based systems.
  • Experience with ML model design, optimization, and HW-SW co-development methodology.
  • ML Model Optimization techniques targeting PPA (Performance, Power, and Area) of neural networks on ARM compute platforms.
  • Adaptability to the fast-moving ML industry and willingness to learn new technology in a very dynamic environment.
  • Salary Range: $191,100-$258,500 per year.
  • Support for accommodations during the recruitment process.
  • Hybrid working environment that supports both high performance and personal wellbeing.
Track Jobs with Teal

Job Search Resources

•
AI Resume Builder
•
Machine Learning Engineer Resume Examples
•
Machine Learning Engineer Cover Letter Examples

Tools

Career Hubs

Guides

Company

© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service