Profile cover
B

AI Embedded Systems ML Engineer

Celtrix AI

Hybrid · $100k – $130k / year · Full-time · Posted 5 months ago

No longer accepting applications

This role is no longer accepting applications

Don't miss the next role like this

Create a free profile with your skills. We match you to new AI roles as they open, and companies hiring can contact you directly.

Create free profile, 2 minutes

Similar open roles you can apply to now:

See all open AI jobs

About the role

PyTorch models that run perfectly on A100 GPUs are useless on an ARM Cortex-M7 with 512KB of RAM. If you understand why — and more importantly, how to fix it — this role is for you. We build AI-powered wearable devices for industrial safety: hard hats that detect fatigue, vests that monitor posture, gloves that predict repetitive strain injuries. The ML challenge is extreme compression. Our models need to run at 30fps on microcontrollers that cost $3 and draw 50mW. You'll take research models from our science team, quantize them, prune them, distill them, and sometimes rewrite them entirely in a framework that supports on-device inference. Then you'll validate that the compressed model still meets our accuracy thresholds in real-world conditions — not just on benchmarks. You'll split time between our San Jose lab (for hardware testing) and remote work. The team is eight engineers across firmware, ML, and cloud, and you'll sit at the intersection of all three.

Responsibilities

  • Compress and optimize ML models for deployment on resource-constrained devices
  • Implement quantization, pruning, and knowledge distillation techniques
  • Validate model accuracy on hardware-in-the-loop test rigs
  • Collaborate with firmware engineers on inference runtime integration
  • Benchmark power consumption and latency against product requirements
  • Maintain a library of optimized model architectures for different device targets

Requirements

  • 3+ years in ML engineering with experience in model optimization
  • Hands-on work with TensorFlow Lite, ONNX Runtime, or similar edge frameworks
  • Understanding of quantization techniques and their accuracy trade-offs
  • Familiarity with embedded systems constraints: memory, power, latency
  • Strong C/C++ skills for performance-critical code paths

Benefits

  • Hybrid role — 2 days in San Jose lab, 3 days remote
  • Health and dental coverage from day one
  • Annual hardware innovation budget of $2,000
  • Patent bonus program
  • Flexible hours around core 10am-3pm PT window

Keep exploring related roles

Use these topic pages to discover more roles in the same hiring cluster.

Categories

AI & Machine Learning Engineers MLOps & AI Infrastructure

Skills

python pytorch tensorflow docker aws

Related remote AI jobs

Nearby roles that share similar skills, categories, or hiring intent.

Browse all jobs

AI Researcher - Pre-training (f/m/d)

Aleph Alpha

Full-time

Research Engineer / Scientist, Pre-training

Anthropic

Full-time

Senior Security Engineer - Detection & Response

Lambda

Full-time Senior

Solutions Engineer, Canada

Glean

Full-time

Senior Data Engineer, Data Analytics

Agility Robotics

Full-time Senior

Sr. Solutions Architect - Travel and Hospitality

Databricks

Full-time Senior

Job Type

Full-time

Level

Mid-Level

Language

English

Salary Range

$100k – $130k / year

AI Expertise

AI & Machine Learning Engineers MLOps & AI Infrastructure

Skills

python pytorch tensorflow docker aws computer vision

Ready to apply for this role?

Create a free talent account.

  • Find AI roles in one place
  • Get AI-matched job recommendations
  • Reply to companies that message you
  • Build your public AI talent profile
Create free account & apply Log in

This role is closed. Create your free profile to apply to similar open roles.

Don't miss the next one: get matched to new roles like this.

Create free profile