Embedded AI Runtime/Kernel Developer at AMBIQ MICRO INC
Austin, TX 78730, USA -
Full Time


Start Date

Immediate

Expiry Date

26 Jun, 25

Salary

0.0

Posted On

27 Mar, 25

Experience

2 year(s) or above

Remote Job

Yes

Telecommute

Yes

Sponsor Visa

No

Skills

C++, Embedded C, Opencl, Keras, Llvm, Cuda, Computer Science, Microcontrollers

Industry

Information Technology/IT

Description

EDUCATION

  • A bachelor’s degree in computer science or a related field requires at least 2 years of relevant experience. A master’s degree or PhD in related topics is highly desirable

REQUIRED SKILLS/ABILITIES

  • Experience writing CPU kernels leveraging vector accelerators such as Arm Helium, Arm Neon, or Intel AVX. Past work with CUDA, OpenCL, or other low-level kernel development environments is a plus.
  • Experience with AI model performance profiling.
  • Experience with embedded C or C++
  • Experience with Keras and Tensorflow (TFLite, TFLite for Microcontrollers).

BONUS QUALIFICATIONS

  • Experience with compiler development
  • Experience with developing for embedded NPUs
  • Past TinyML/EdgeAI involvement or experience
  • Experience developing and optimizing for TFLite for Microcontrollers
  • Experience with model-to-binary compilers (IREE, MicroTVM, etc)
  • Experience with ONNX, TOSA, Jax, LLVM, and/or MLIR
  • Experience with optimizing for heterogeneous AI compute (e.g., CPU+NPU+DSP)
Responsibilities
  • Optimize embedded AI runtimes such as Tensorflow Lite for Microcontrollers to utilize Ambiq’s hardware products efficiently.
  • Develop advanced inference performance profiling tools to help customers identify optimization targets and solutions.
  • Develop novel, ahead-of-time AI model inference compilers to achieve better power, latency, and memory performance incorporating state-of-the-art pruning and quantization techniques.
  • Develop training-side tools and libraries to help AI developers identify neural architectures that optimally run on Ambiq’s platforms.
  • Publish and maintain these tools, including documentation and other assets our customers need to bootstrap their internal AI features.
  • Socialize their achievements via conferences, meetups, workshops, and publications.
    Requirements:
Loading...