Albert Invent

Invent the future, faster.

Staff AI/ML Platform Engineer

Platform EngineerPlatform EngineerFull TimeRemoteTeam 51-200Since 2022H1B No SponsorCompany SiteLinkedIn

Location

California

Posted

34 days ago

Salary

Not specified

Bachelor Degree7 yrs expEnglishAWSCloudDockerDynamo DBKubernetesNo SQLPythonRayRedis

Job Description

• You'll own the APIs, data pipelines, and workflow orchestration that power our AI products—from real-time model inference to long-running optimization pipelines. • This role sits at the intersection of backend engineering and data engineering: you'll build the services that serve up models, manage workflows, and connect AI capabilities to the structured data that makes them useful. • You'll work closely with our Active Learning and LLM/Agents team leads, translating their product vision into scalable, production-grade systems. • The infrastructure you build will power model playgrounds for chemists, inverse design pipelines that optimize experiments across high-dimensional spaces, and orchestrated agent workflows that reason through complex scientific problems. • Design and build high-performance Python APIs that serve models, manage workflows, and expose AI capabilities to the broader platform • Architect backend services for scalability, reliability, and low latency • Build integrations between AI/ML systems, graph databases, and external data sources • Build and maintain long-running workflow pipelines using Ray and Temporal. • Design orchestration patterns for multi-step agent pipelines, batch inference, and numerical optimization workflows • Ensure fault tolerance, graceful degradation, and efficient resource utilization. • Architect and maintain data pipelines that feed AI/ML workflows • Work with Neptune (graph), Redis, DynamoDB, and other data stores to enable efficient data access patterns • Implement observability including logging, metrics, tracing, and alerting • Own system reliability—troubleshoot issues, conduct post-mortems, and continuously improve. • Design CI/CD pipelines and promote automation best practices.

Job Requirements

  • Deep expertise in Python backend development and building production APIs
  • Experience designing and operating data pipelines and workflow orchestration systems
  • A builder's mindset—you want to create foundational systems that others build on
  • Genuine curiosity about how your work enables scientific discovery
  • A commitment to rigor: AI makes mistakes confidently, and our customers won't accept hand-waving—neither should we
  • A degree in Computer Science or a related field with 7+ years of industry experience (Bachelor's) or 5+ years (Master's or PhD) in software engineering
  • Advanced proficiency in Python including async programming and performance optimization
  • Experience building and maintaining REST APIs using FastAPI or similar frameworks
  • Experience with workflow orchestration tools (Ray, Temporal, or similar)
  • Strong background in data engineering: pipelines, transformations, and working with diverse data stores
  • Experience with cloud platforms (AWS preferred) and containerization (Docker, Kubernetes)
  • Familiarity with graph databases, key-value stores, or other NoSQL systems (Neptune, Redis, DynamoDB a plus)
  • Track record of operating production systems at scale.

Benefits

  • We care about you.
  • We love distributed teams.
  • We value diversity.

Related Categories

Related Job Pages

More Platform Engineer Jobs

ML Platform Engineer

Afresh

The smartest solution for fresh

Platform Engineer34 days ago
Full TimeRemoteTeam 51-200Since 2017H1B Sponsor

ML Platform Engineer enhancing Afresh's machine learning platform.

Python
Alabama + 20 moreAll locations: Alabama, California, Colorado, Florida, Illinois, Kentucky, Montana, Nevada, New Jersey, New York, North Carolina, Oregon, Massachusetts, Michigan, Missouri, Pennsylvania, Texas, Utah, Virginia, Washington, Wisconsin

Platform Engineer

Curri

Transforming the way construction and industrial supplies are delivered.

Platform Engineer35 days ago
Full TimeRemoteTeam 51-200H1B No Sponsor

Platform Engineer building internal platform for Curri's engineering teams

AWSEC2KubernetesTerraformTypeScript
California

Senior Platform Engineer

Pondurance

Delivering personalized, 24/7 MDR services that grow with your organization.

Platform Engineer35 days ago
Full TimeRemoteTeam 51-200H1B No Sponsor

Senior Platform Engineer supporting and enhancing Pondurance’s MDR Data Pipeline

CloudDockerLinuxPythonSaltStackTerraform
Virginia

Power Platform Developer

LMI

Innovation at the Pace of Need™

Platform Engineer35 days ago
Full TimeRemoteTeam 1,001-5,000Since 1961H1B Sponsor

SharePoint Developer supporting Army Special Operations with SharePoint modernization

SQL.NET
United States
$109.3K - $189.0K / year