High-Performance ML Inference Engineer for Diffusion Models

Reactor; San Francisco (CA); On-site; USD 120,000 - 160,000; Full time

Benefits offered by this job

Visa sponsorship

Relocation support

Generous health, dental, and vision coverage

Job summary

Reactor is looking for an experienced ML Inference Engineer with deep expertise in high-performance ML engineering. This role focuses on optimizing the performance of generative media models, contributing to Reactor's competitive edge.

The ideal candidate will drive model performance for diffusion models, collaborate with partner teams, and implement optimizations. A Bachelor's degree in a related field is required, along with strong knowledge of GPU hardware and modern ML optimization techniques. The position offers competitive salary and meaningful equity.

Qualifications

• Strong foundation in systems programming, with a track record of identifying and resolving bottlenecks.
• Working knowledge of GPU hardware (NVIDIA).
• Strong understanding of transformer architectures and modern ML optimization techniques.

Responsibilities

• Drive frontier position on model performance for diffusion models.
• Design and implement a in-house inference runtime.
• Profile and benchmark model performance to identify computational bottlenecks.

Skills

Deep expertise in PyTorch

TensorRT

Advanced serving architectures

Model compilation

Quantization

Education

Bachelor's degree in Computer Science, Electrical Engineering, or related

Tools

Nsight

ONNX Runtime

Job description

Reactor is looking for an experienced ML Inference Engineer with deep expertise in high-performance ML engineering. This role focuses on optimizing the performance of generative media models, contributing to Reactor's competitive edge.

The ideal candidate will drive model performance for diffusion models, collaborate with partner teams, and implement optimizations. A Bachelor's degree in a related field is required, along with strong knowledge of GPU hardware and modern ML optimization techniques. The position offers competitive salary and meaningful equity.