Staff Machine Learning Engineer

Workman Dental Labs

Mountain View, CA, United States

4 days ago

Apply on www.juju.com

Prepare application

Role details

Contract type

Permanent contract

Employment type

Full-time (> 32 hours)

Experience level

Expert

Experience required

6 years minimum

Compensation

$218,400 - $327,600

Working hours

Regular working hours

Job source

www.juju.com

Tech stack

Artificial Intelligence Computer Vision Game Engine Python (Programming Language) Machine Learning Language Modeling Software Architecture NeRF (Neural Radiance Fields) Neural Rendering Pytorch Discretization Low Latency

+3 more

ONNX (Open Neural Network Exchange) Format Stable Diffusion GPT

Job description

Establish technical vision and roadmap for computer vision and multi-modal AI (transformers, diffusion models, vision-language models)
Drive design of systems for image/video understanding, generation, segmentation, detection, and multi-modal reasoning
Make architectural decisions balancing quality, latency, and cost across cloud, server, and on-device targets
Own the full pipeline from research prototypes to production: training, fine-tuning, distillation, serving
Apply compression, quantization, pruning, and knowledge distillation for constrained environments Unique
Lead and mentor ML engineers; define and track KPIs for model quality, accuracy, latency, memory, and cost

Requirements

6+ years ML engineering with significant depth in computer vision and/or multi-modal modeling
Production experience with transformer-based and diffusion vision models (ViT, CLIP, Stable Diffusion, DETR, SAM)
Strong full model lifecycle knowledge: data curation, training, evaluation, serving at scale
Familiarity with efficient attention, diffusion samplers, multi-modal fusion, and vision-language alignment Unique
Strong Python and PyTorch skills; proven technical leadership track record
Preferred: world-model/video-generation/neural rendering pipelines (NeRF, 3DGS), on-device deployment (CoreML, TFLite, ONNX), real-time graphics or game engine background Common

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on www.juju.com

Prepare application

Good distractions

Loading talks and stories from around this role…