Senior Optimization / Algorithms Engineer

EPAM Systems Inc

United States

On-site

USD 130,000 - 190,000

Full time

4 days ago
Be an early applicant
Application generator

Stand out for this role — generate a tailored resume and cover letter in about a minute.

Get past ATS filters

Job summary

EPAM Systems is seeking a Senior/Lead Optimization/Algorithms Engineer to design and build the combinatorial optimization core powering topology-aware TPU scheduling. You will translate scheduling constraints into solvable mathematical models and drive sub-second performance at scale.

Responsibilities include developing a solver core for TPU slice placement, building MIP/CP-SAT representations of multi-host constraints, profiling latency, and integrating Kubernetes scheduling considerations into

Qualifications

  • Strong background in combinatorial optimization, operations research, or algorithms engineering.
  • Experience with MIP/CP-SAT solvers (OR-Tools, Gurobi, CPLEX).
  • Proficiency in Python and/or Golang.
  • Understanding of graph theory and constraint satisfaction problems.
  • Familiarity with performance profiling and latency-critical systems.

Responsibilities

  • Develop the combinatorial solver core for TPU slice placement.
  • Build MIP/CP-SAT models representing multi-host slice constraints.
  • Perform latency profiling to ensure sub-second solve times at scale.
  • Translate Kubernetes CRD/Pod spec constraints into abstract graph/optimization models.
  • Collaborate with AI/ML Performance Engineers to incorporate empirical hardware coefficients into the cost model.

Skills

Combinatorial optimization
Operations research
Algorithms engineering
Python
Golang
Graph theory
Latency profiling

Tools

OR-Tools
Gurobi
CPLEX

Job description

We are looking for a Senior/Lead Optimization/Algorithms Engineer to design and build the combinatorial optimization core that powers topology-aware TPU scheduling, translating scheduling constraints into solvable mathematical models.

Responsibilities
  • Develop the combinatorial solver core for TPU slice placement
  • Build MIP/CP-SAT models representing multi-host slice constraints
  • Perform latency profiling to ensure sub-second solve times at scale
  • Translate Kubernetes CRD/Pod spec constraints into abstract graph/optimization models
  • Collaborate with AI/ML Performance Engineers to incorporate empirical hardware coefficients into the cost model
Requirements
  • Strong background in combinatorial optimization, operations research, or algorithms engineering
  • Experience with MIP/CP-SAT solvers (e.g., OR-Tools, Gurobi, CPLEX)
  • Proficiency in Python and/or Golang
  • Understanding of graph theory and constraint satisfaction problems
  • Familiarity with performance profiling and latency-critical systems
  • Nice to have Prior exposure to Kubernetes scheduling internals
  • Experience with distributed systems optimization
Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Lead TPU Scheduling & Optimization Architect
Lead TPU Scheduling & Optimization Architect

EPAM Systems Inc • United States

Remote
USD 130,000 - 190,000
Senior/Lead AI/ML Performance Engineer
Senior/Lead AI/ML Performance Engineer

EPAM Systems Inc • United States

Remote
USD 150,000 - 230,000
NPU Kernel/Operator Engineer
NPU Kernel/Operator Engineer

Black Sesame Technologies Inc • San Jose (CA)

On-site
USD 120,000 - 160,000
Member of Technical Staff, GPU Kernels
Member of Technical Staff, GPU Kernels

SF Tensor • San Francisco (CA)

On-site
USD 180,000 - 240,000
Cloud DevOps Engineer (GCP)
Cloud DevOps Engineer (GCP)

EPAM Systems Inc • United States

Remote
USD 150,000 - 210,000
Optimization Engineer
Optimization Engineer

ANAUTICS INC • Oklahoma City (OK)

On-site
USD 80,000 - 110,000
HPC Infrastructure Engineer
HPC Infrastructure Engineer

Arcadia • San Francisco (CA)

On-site
USD 180,000 - 260,000
Senior AI/ML Performance Engineer - TPU Benchmarking
Senior AI/ML Performance Engineer - TPU Benchmarking

EPAM Systems Inc • United States

Remote
USD 150,000 - 230,000
Machine Learning Engineer- Inference Optimization | Experienced Hire
Machine Learning Engineer- Inference Optimization | Experienced Hire

Susquehanna International Group, LLP • Bala Cynwyd (PA)

On-site
USD 110,000 - 150,000
Software Engineer, Compute Foundations
Software Engineer, Compute Foundations

Linuxcareers • San Francisco (CA), Northern (KY)

On-site
USD 210,000 - 270,000