Level

Brain Co

AI Platform Engineer, Backend Capabilities

AI in this role

mlflowweights-and-biasesdatabricks
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

How we rate this

AI Platform Engineer, Backend Capabilities at Brain Co rates 90 out of 100 for how much of the daily work is AI. That makes it Builds AI (AI Level 4 of 4). The level is about AI in the job, not seniority.

Classification

Builds AI. The job is building AI systems.

  1. ●●●● Builds AI80 to 100
  2. ●●●○ Works on AI60 to 79
  3. ●●○○ Uses AI40 to 59
  4. ●○○○ Little AI0 to 39

Levels come from how often the tools, models and workflows of the role are named in the posting itself. Open the description and count.

Prepare for this job

A free preview built only from this posting: what it asks for, what you could be asked in an interview, and how to adjust your resume.

Skills and AI tools this role asks for

MlflowWeights And BiasesDatabricks

Questions you could be asked

  1. What's a project where you used Mlflow hands-on?
  2. Walk me through how you've used Weights And Biases in your day-to-day work.
  3. What are the limits of Databricks that you've run into, and how did you work around them?
  4. How would you decide a model or AI system is ready to ship?
  5. Tell me about a time a model underperformed in production. How did you find out, and what did you change?

Adapt your resume

  • List these exact terms on your resume: Mlflow, Weights And Biases, and Databricks. An applicant tracking system matches the wording, not the idea.
  • Attach one line of real, concrete experience to at least one of them — a tool named with nothing behind it rarely survives a human read.
  • Lead with what you built, trained or shipped — this role is judged on the AI system itself, not the tools around it.

Want an expert to read your CV for this job?

Free. Send your CV and the role you want next. We reply by email within 2 to 4 business days.

Get a free CV review

Get new remote AI jobs (Builds AI ●●●●) by email

One email a week with the new remote AI jobs (Builds AI ●●●●), each rated for how much AI is in the work. No recruiter spam, unsubscribe in one click.

Free. One email a week. Unsubscribe in one click.

Similar roles

Software Engineering roles that build AI, at other companies.

What kind of AI work fits you?

Answer 12 practical questions in about three minutes. Get a simple profile, the work it points to, and live roles to explore next.

Find my next step

More jobs at Brain Co

Related searches

Same AI level