Staff Software Engineer, Environments Infrastructure

Job summary

ABOUT Ant*** Ant***’s mission is to create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for our users and for society as a whole. Our team is a quickly growing group of committed researchers, engineers, policy experts, and business leaders working together to build beneficial AI systems.

Benefits

Qualifications

• Deep expertise in Python, including static typing, safe async and concurrency patterns, and writing performant code
• Experience designing or operating stateful concurrent or distributed systems, and reasoning carefully about failure, retires, idempotency, and consistency
• A habit of verification: you measure before you conclude, and you build the checks that let a system show it's correct
• Experience working productively in large, evolving, or research-style codebases that you didn't originally write
• PREFERRED
• Experience building infrastructure, tooling, or frameworks for machine learning research or RL workflows, and familiarity with agentic systems or LLM training pipelines
• Experience building agent frameworks, orchestration engines, or multi-agent systems, including checkpoint and restore, replay, and coordination of long-running stateful processes
• Experience using AI coding
• for the position
• Location-based hybrid policy: Currently, we expect all staff to be in one of our offices at least 25% of the time.
• However, some roles may require more time in our offices.
• Visa sponsorship: We do sponsor visas!

Responsibilities

• That includes the frameworks researchers use to build environments and the infrastructure responsible for running them.
• You'll embed with research teams, get up to speed on how they work, and design the frameworks and APIs that let them move faster, building systems the team can understand, own, and maintain themselves.
• You'll be a strong fit if you have deep expertise in Python, a refined sense of taste for API and framework design, and good intuition for how complex systems fail, especially silently.
• Design widely used APIs, frameworks, and abstractions that other engineers and researchers build on, making correct usage the default and ruling out entire classes of errors structurally
• Own the platform layers that sit beneath every environment, including the agent runtime
• Build the tooling that lets environment owners understand, debug, and maintain their environments in production without needing an infrastructure engineer in the loop
• Embed with research teams on a rotational basis, work directly in their codebases without slowing down the research they support, and transfer ownership when you rotate off
• Anticipate silent failure modes and prevent them structurally through type safety, well-designed invariants, targeted testing, and refactors that reduce the room for correctness issues
• Drive adoption of new frameworks across the organization, including deprecations and cutovers
• Help define the engineering standards, review practices, and design patterns for a new team, and mentor researchers and engineers in adopting them
• MINIMUM

Skills

Degrees

Work schedule

Industry

Company size