Software Engineer: ML Optimization

Home > Python programming jobs

Software Engineer: ML Optimization in USA new

Generalist
Full time
Email

USA
11/06/2026
Report

Boston, MA

Responsibilities

Own optimizations across machine learning training and on-robot inference stacks.
Implement ML, hardware, and software changes that deliver step-function performance gains.
Optimize GPU execution, CUDA kernels, ML architectures, network bottlenecks, CPU performance, NVLink communications, and framework-level inefficiencies.
Investigate and improve performance across PyTorch, NumPy, and Python.

Requirements

Proficiency with current machine learning techniques for training and inference optimization in transformer- and diffusion-based architectures.
Willingness to pursue optimization opportunities across kernels, models, networks, CPUs, communications, and software frameworks.

Salary: $260k - $350k/yr

Generalist

Generalist builds general‑purpose robots powered by large‑scale AI, focusing on vision‑language‑action and embodied multimodal models for real‑world tasks in industry and homes. The company develops robotics platforms and software and is privately held. It was founded in 2024, is headquartered in...

Similar positions

Staff Software Engineer

Crunchyroll, LLC
Full time

Others
11/06/2026

Hyderābād, India

Software Engineer, Cyber Physical Systems

Kairos Power
Full time

USA
11/06/2026
Salary: $92k - $109k/yr

Alameda, CA

Senior Software Engineer (CCC Auto

EvolutionIQ
Full time

Remote
11/06/2026
Salary: $180k-$225k

United States

Forward Deployed Engineer - Software Engineer

ElevenLabs
Full time

Remote
11/06/2026

Argentina

Software Engineer - Computer Vision

Verkada
Full time

USA
11/06/2026

San Mateo, CA