ML Engineering Intern: Build & Optimize Cutting-Edge Models

Matrice AI; United States; Remote; USD 150,000 - 190,000; Full time

Job summary

Matrice AI is seeking a highly skilled ML Engineer to design and optimize training and inference systems for cutting-edge vision-language and multi-modal models. You will implement neural architectures from first principles and ensure efficiency on edge and cloud platforms.

You’ll own end-to-end model implementations, deepen your math understanding, and trade off performance with memory and speed. Expect a hands-on, high-ownership role in a fast-moving field.

Qualifications

• Expert-level proficiency in Python across problem domains.
• Strong CS fundamentals: memory management, multi-threading, distributed systems.
• Extreme ownership and accountability for code quality and system performance.
• Willingness to learn the mathematics behind models, not just use libraries.

Responsibilities

• Design training and inference pipelines for state-of-the-art models.
• Write performant inference code bridging research and production.
• Implement neural architectures from first principles using Python/NumPy.
• Debug and optimize for speed, memory, and accuracy.

Skills

Python

Multi‑threading

Memory management

Distributed systems

Code quality

Ownership

Tools

NumPy

PyTorch

Job description

Matrice AI is seeking a highly skilled ML Engineer to design and optimize training and inference systems for cutting-edge vision-language and multi-modal models. You will implement neural architectures from first principles and ensure efficiency on edge and cloud platforms.

You’ll own end-to-end model implementations, deepen your math understanding, and trade off performance with memory and speed. Expect a hands-on, high-ownership role in a fast-moving field.