ML Infrastructure Engineer

ABOUT THE ROLE

This is a hands-on ML Infrastructure Engineer role at an early-stage enterprise AI company building a context and data governance layer that makes AI agents reliable in production. You will own the inference and model-serving infrastructure end to end, ensuring agents run fast and reliably at increasing concurrency. The work is squarely production-focused with real-world impact across regulated industries like insurance, banking, healthcare, and asset management.

WHAT YOU'LL DO

• Design, build, and scale inference and model-serving infrastructure from the ground up through production deployment.
• Optimize systems for latency, throughput, and reliability under high concurrency.
• Collaborate closely with ML and infrastructure teams to ensure seamless integration and surface performance bottlenecks.
• Drive solutions to infrastructure challenges across a fast-moving, cross-functional team.

WHAT WE'RE LOOKING FOR

• 5 or more years building and operating machine learning inference systems, model-serving platforms, or ML infrastructure in production environments.
• Hands-on experience designing and scaling inference-serving systems using tools such as TensorFlow Serving, TorchServe, Triton, KServe, or equivalent custom solutions.
• Strong distributed systems fundamentals, including containerization and orchestration with Docker and Kubernetes.
• Proficiency with monitoring and observability tooling for production systems, such as Prometheus, Grafana, or distributed tracing frameworks.
• Experience deploying and managing ML workloads on cloud platforms (AWS, GCP, or Azure).
• Proficiency in at least one systems or backend language: Python, Go, Rust, C++, or Java.
• Comfort collaborating across both ML and infrastructure disciplines in a fast-paced environment.
• Nice to have: experience with knowledge graphs, semantic search, or graph databases; real-time or low-latency inference systems; agentic or multi-step AI pipelines; enterprise data integration or pipeline infrastructure.

LOCATION

On-site in San Mateo, California, United States. Visa sponsorship is not available for this role.