Begaluru, India
Full Time

ML Compiler Engineer

Job Responsibilities

At Avataar Reality labs, we are pushing the boundaries of what is possible in AR/VR and enabling the consumer shift to metaverse at scale. We are building the neural engine to empower the creation, rendering and manipulation of AR/XR content across devices. Our work involves solving challenging problems at the intersection of Computer Vision, Graphics and Deep Learning.

As a ML Compiler Engineer, you will be working on building the framework for optimizing and deploying computer vision models across devices, making best use of modern parallel environments.

Job Requirements

  • Proficiency in C/C++
  • Strong programming experience in CUDA or OpenCL.
  • Strong grasp of CPU and GPU performance challenges and optimization techniques.
  • Experience working with at least one deep-learning library (e.g., PyTorch, Jax,TensorFlow, Caffe2). This experience should include optimization of new algorithms.

Preferred Requirements:

  • Experience working with WebAssembly.
  • Experience in shipping computer vision or image processing products to customers

Career Description

We are looking for an ML Compiler Engineer to join Avataar's Reality Labs team

Apply now