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

Lead Software Engineer

JPMorgan Chase
Full time

USA
11/06/2026
Salary: Competitive

Plano, TX

Senior Software Engineer, Engineering Enablement

Motional
Full time

Others
11/06/2026

Singapore, Singapore

Forward Deployed Engineer - Software Engineer

ElevenLabs
Full time

UK
11/05/2026

Remote

Senior Software Development Engineer in

Varicent
Full time

Canada
11/05/2026
Salary: C$101k-C$132k

Remote

Principal Software Engineer/Developer

Fidelity
Full time

USA
11/05/2026

Merrimack, NH