Machine Learning Software Engineer


Job Overview

Key Qualifications

  • Strong Python programming skills.
  • Understanding of data structures, software design principles and algorithms.
  • Experience with Python scientific libraries like numpy, scipy.
  • Experience with one or more deep learning framework such as Tensorflow, PyTorch or Keras.
  • Experience developing machine learning solutions.
  • Understanding of compute hardware design, CPU and GPU architectures.
  • Familiarity with GPU programming, shader programming, CUDA, Metal or OpenCL.
  • Experience with C++, Objective-C, Swift.
  • iOS development experience.
  • Solid math skills: linear algebra, computational science and optimization.
  • Publication record in ML conferences is a plus.
  • Description

    We’re looking for strong software engineers to build a next generation Deep Learning framework geared towards fueling on-device machine learning capabilities and emerging innovations. You’ll be part of small team of developers and deep learning experts working in the area of device-aware neural network optimization and neural architecture search. We’re looking for candidates with polished coding skills, passionate about machine learning, computational science and hardware. Any background and experience level is acceptable. In exchange, we offer respectful work environment, flexible set of responsibilities, access to world class experts and growth opportunities in one of the best companies in the world.Design and develop APIs for common and emerging deep learning primitives: layers, tensor operations, optimizers and more specific hardware features. Implement efficient tensor operations and DNN training algorithms. Train and evaluate DNNs for the purpose of benchmarking neural network optimization algorithms. Our framework reduces latency and power consumption of neural networks found in many Apple products. Perform research in emerging areas of efficient neural network development including quantization, pruning, compression and neural architecture search, as well as novel differentiable compute primitives. We encourage publishing novel research at top ML conferences.

    Education & Experience

    PhD or MS CS/CE/EE (or equivalent) in the area of computer science, computer vision, imaging science, optics, is helpful

    View More
    Job Detail
    Shortlist Never pay anyone for job application test or interview.