AI/ML Platform Engineer: Real-Time Models & Pipelines

F-Prime Capital; United States; On-site; USD 150,000 - 210,000; Full time

Job summary

Albert Invent is building AI-powered chemistry platforms and needs an AI/ML Platform Engineer to own APIs, data pipelines, and workflow orchestration. You will ship production-grade services that enable real-time model inference and scalable multi-step pipelines for researchers.

You'll collaborate with ML researchers and data scientists to translate product vision into robust infrastructure, with opportunities to influence architecture and adopt new technologies across the platform.

Qualifications

• 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

Responsibilities

• Design and build high-performance Python APIs that serve models, manage workflows, and expose AI capabilities.
• 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, Redis, DynamoDB, and other data stores to enable efficient data access patterns.
• Build the connectors and transformations that give AI systems access to clean, structured, trusted data.
• 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.
• Partner with ML researchers, data scientists, and product engineers to understand requirements and deliver production-ready infrastructure.
• Collaborate closely with Active Learning and LLM/Agents team leads to align platform capabilities with product needs.
• Contribute to architectural decisions that shape how AI gets built and shipped at Albert.

Skills

Python backend

REST APIs

Async programming

Performance optimization

Data engineering

Cloud platforms

Docker

Kubernetes

NoSQL databases

Production systems

Ray

Temporal

Education

Bachelor's degree in Computer Science or related field

Master's or PhD (preferred)

Tools

Ray

Temporal

FastAPI

AWS

Neptune

Redis

DynamoDB

Docker

Kubernetes

Job description

Albert Invent is building AI-powered chemistry platforms and needs an AI/ML Platform Engineer to own APIs, data pipelines, and workflow orchestration. You will ship production-grade services that enable real-time model inference and scalable multi-step pipelines for researchers.

You'll collaborate with ML researchers and data scientists to translate product vision into robust infrastructure, with opportunities to influence architecture and adopt new technologies across the platform.