AI Platform Engineer, Backend Capabilities
At a glance
Mid-level ML Platform & Ops role at Brain Co. San Francisco · full-time · no salary stated · we estimate $209,400–$315,250.
Brain Co. has not published a salary for this role. The figure below is our estimate, not theirs.
aimlroles summary, based on the employer's posting.
What you'll do
- Design, deploy, and maintain backend services and data flows behind AI products across their full lifecycle
- Develop infrastructure for ML experiment records, stored artifacts, and automated model training and evaluation
- Build fault-tolerant services with strong observability to meet customer uptime and latency expectations
- Create scalable APIs and event-driven components, then improve their latency, throughput, and cost
- Coordinate with product, ML, infrastructure, and customer-facing teams to clear obstacles to shipping AI products
What you bring
- At least two years delivering production backend services or platform systems that can handle growth
- Strong command of Python, Go, Rust, TypeScript, C++, or a similar language
- A firm grasp of distributed systems concepts such as consistency, availability, failure handling, retries, and idempotency
- Experience creating internal infrastructure or developer services with straightforward, well-documented APIs
- You have owned services with uptime commitments and taken on monitoring, incident response, and on-call duties
Who this fits
This role suits a backend engineer who can turn loosely defined problems into production systems and balance immediate product needs with platform health. You’ll collaborate with product, ML, infrastructure, and customer-facing teams. The position is hybrid in San Francisco.
From the employer
Our Mission
Rebuild how the world works, to make institutions work better for the people they serve.
About Brain Co.
Brain Co. builds AI-native operating systems for large, regulated institutions. Each system is built for a specific industry, powered by agents that push real workflows forward. Underneath it all is Atlas, our proprietary platform that keeps customers in control, secure by design, and never locked into one model.
Why Now
Brain Co. is entering its next phase of production deployments on a national scale with an elite team built from Palantir, Google, Meta, and Nvidia, and a growing footprint across government, insurance, health, and financial services.
Joining now means shaping both the company and a new category of applied AI. Every project here ships to production and is expected to create measurable customer value and impact.
You'll work alongside exceptional peers on some of the hardest problems in applied AI. It’s the kind of work you'll still be proud of in ten years from now.
About the Role:
As an AI Platform Engineer at Brain Co., you will build the shared backend services and technical capabilities that help us scale our AI products quickly, safely, and reliably. You will take ambiguous product and technical challenges, turn them into clear system designs, and ship robust platforms that support real-world AI applications for the world’s most important institutions. You will own backend services from design through production, working closely with teams across Product, ML, and Infrastructure on system design and technical tradeoffs.
What You’ll Work On:
Design, build, and operate the platform backend services and data pipelines that power Brain Co.'s AI products. You will own the full lifecycle: from initial architecture and implementation to deployment and long-term maintenance.
Build the critical systems that accelerate our AI product development. This includes designing scalable solutions for ML experiment tracking, artifact management, and automated training and evaluation pipelines.
Engineer highly available, fault-tolerant systems with deep observability. Your architectures must be robust enough to meet the strict uptime and latency SLAs demanded by our enterprise and government clients.
Design modular and scalable architectures, clean APIs (REST, gRPC), and event-driven services with a long-term platform mindset. Continuously profile systems to optimize for latency, throughput, and costs.
Partner with Product, ML, Infrastructure, and customer-facing teams to build shared platform capabilities that remove bottlenecks and reduce the time it takes to ship new AI products to our clients.
You Might Be a Great Fit If You
You bring 2+ years of experience building and scaling production backend services or platforms, with strong proficiency in Python, Go, Rust, TypeScript, C++, or similar languages.
You understand what happens under the hood. You have strong fundamentals in distributed systems, including consistency, availability, failure modes, retries, and idempotency.
You excel at breaking down complex, open-ended problems into clear technical designs, moving from first principles to production-ready systems with both speed and rigor.
You treat our internal ML and product teams as your primary customers. You have experience building shared infrastructure, internal platforms, or developer-facing services, with a focus on intuitive, well-documented APIs that help these teams build and ship effectively.
You have a track record of owning services with real uptime expectations. You design for observability from day one, using metrics, logging, and tracing, and take responsibility for incident response and on-call support for the systems you build.
You treat the platform as your own. You care about the end-to-end lifecycle of your systems and make pragmatic tradeoffs between immediate product needs and long-term platform health.
Bonus Points For:
Experience with AI/ML platforms or inference systems, including integrating or managing ML experiment tracking, artifact management, and model registries (e.g., Weights & Biases, MLflow, ClearML) or orchestrators (e.g., Ray, Flyte, Kubeflow).
Experience designing and operating high-throughput data pipelines and resilient asynchronous workflows using durable execution engines (e.g., Temporal), streaming platforms (e.g., Kafka), or distributed compute frameworks (e.g., Spark, Flink).
Familiarity with Kubernetes, cloud-native service deployment, or multi-tenant architectures.
Experience navigating strict compliance frameworks (e.g., SOC2, FedRAMP) or building systems for highly regulated, air-gapped, or on-premise environments.
Why Join Us:
Work alongside senior engineers from Tesla, DeepMind, Databricks, and other top engineering orgs.
Ship fast, learn constantly, and see your work protect production systems used by millions.
Earn competitive compensation and meaningful equity in a high-growth company.
Benefits
Competitive salary plus equity
Daily lunches
Commuter benefits
401(k)
Medical, Dental, and Vision
Unlimited PTO