Senior Optimization / Algorithms Engineer

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

Benefits

• Opportunity to work in a fast-paced, agile, software engineering culture
• Comfortable modern office in Prague 7, with support of hybrid or fully remote mode
• Benefit program (5 weeks of vacation, paid sick days, paid days off for special occasions, meal vouchers, flexi pass, Prague city public transport annual coupon, multisport cards, optional contribution to pension fund, health insurance for family member)
• EPAM Employee Stock Purchase Plan (ESPP) (subject to certain eligibility requirements)
• English language courses
• Czech language courses upon request
• Referral bonuses for recommended candidates
• Mobile Phone Tariff’s program for managerial-level candidates
• Great learning and development opportunities, including in-house professional training, career advisory and coaching, sponsored professional certifications, well-being programs, LinkedIn Learning Solutions and much more

• Opportunity to work in a fast-paced, agile, software engineering culture
• Benefit program (5 weeks of vacation, 5 paid sick days, meal vouchers, cafeteria and recreation bonuses, reimbursement of glasses, contribution to pension fund)
• Referral bonuses for recommended candidates
• English language courses
• Great learning and development opportunities, including in-house professional training, career advisory and coaching, sponsored professional certifications, well-being programs, LinkedIn Learning Solutions and much more