Skip to content

Machine Learning Engineer, Infrastructure

Build and improve ML systems and data pipelines infrastructure to support modeling engineers. Requires 2+ years experience, bachelor's in CS/math/sciences, strong coding in Python/Go/Java/C++, and production ML infra expertise.

175k – 270kUnited StatesML EngineeringHybrid2+ YOE

About the role

You will:

  • Design, build, and improve ML systems and Data pipelines infrastructure
  • Work with and enable other ML engineers focused on modeling
  • Write robust code that's easy to read, maintain, and test
  • Mentor more junior engineers, or learn from battle-tested ones

About you:

  • 2+ years of experience
  • BA/BS in computer science, math, sciences, or a related degree
  • Proven ability to design, build, and ship production-ready software, ideally around AI/ML infrastructure (Pipelines, Serving etc)
  • Strong coding skills (Python, Go, Java, C++, ...)
  • Thrive in a customer-focused, tight-knit and cross-functional environment - being a team player and willing to take on whatever is most impactful for the company is a must
  • A proactive and positive attitude to lead, learn, troubleshoot and take ownership of both small tasks and large features

Compensation & Benefits

The standard base salary range for this position is $175,000 - $270,000 annually. Compensation offered will be determined by factors such as location, level, job-related knowledge, skills, and experience. Certain roles may be eligible for variable compensation, equity, and benefits.

We offer a comprehensive benefits package including competitive compensation, Medical, Vision, and Dental coverage, generous time-off policy, and the opportunity to contribute to your 401k plan to support your long-term goals. When you join, you'll receive a home office improvement stipend, as well as an annual education and wellness stipends to support your growth and wellbeing. We foster a vibrant company culture through regular events, and provide healthy lunches daily to keep you fueled and focused.

Skills

PythonGoJavaC++Ml PipelinesData PipelinesAI InfrastructureRAGGenerative AIKubernetes

Similar roles

ML Engineering jobs

ML Engineer, Generative Video

Build and scale video generation models at Mirage. Optimize large-scale multimodal models for training efficiency, low-latency inference, and production deployment using PyTorch, CUDA, and distributed systems. 2+ years industry experience required; on-site in NYC.

175k – 275kNew York, NYML EngineeringOn-site2+ YOECUDAFsdp

Machine Learning Engineer - 3D Sensor Simulation

Develops and optimizes ML-driven 3D sensor simulations (cameras, lidar, radar) using GenAI and graphics techniques to generate realistic synthetic data for AV testing. Requires 2+ years experience with neural rendering (NeRFs, Gaussian Splatting), PyTorch/TensorFlow, Python/C++, and 3D math.

176k – 257kFoster City, CAML EngineeringHybrid2+ YOEC++Nerf

Machine Learning Engineer - 3D Sensor Simulation

Develops and optimizes 3D sensor simulations using GenAI/ML and graphics techniques for realistic camera, lidar, and radar data in autonomous vehicles. Requires 2+ years in neural rendering like NeRFs/Gaussian Splatting, PyTorch/TensorFlow, Python/C++.

173k – 244kSeattle, WAML EngineeringHybrid2+ YOEC++Nerfs

Software Engineer, AI

Convex is seeking a Software Engineer to build and architect their AI platform, focusing on creating seamless agentic development experiences. This role involves working across APIs, plugins, and AI framework integrations, identifying gaps, and prioritizing projects to enhance the platform.

170k+San Francisco, CAML EngineeringOn-siteEntry levelAPIsTypeScript

Data Scientist, Core Data - PhD (2026)

Research-minded Data Scientist building experimentation platforms, ML analytical systems, and causal inference models for data-driven decisions at Figma. Requires PhD in quantitative field, SQL/Python fluency, and strong stats/ML foundation.

170k – 178kNew York, NY +1ML EngineeringRemoteEntry levelRSQL