Build scalable machine learning platforms and production MLOps workflows that help Amgen scientists develop, deploy, and manage AI/ML solutions for drug discovery and development.
Model Serving Engineer
π Description
- Optimize Python inference code for concurrency and performance, including threading, multiprocessing, async execution, and long-running workloads.
- Partner with research teams to bring evolving model architectures into production and understand their computational behavior.
- Design distributed inference and asynchronous execution systems, including orchestration and worker coordination.
- Own the Triton serving layer, including model packaging, configuration, and execution in the production pipeline.
- Tune hardware utilization and serving performance across CPU, GPU, memory, networking, batching, and concurrency.
- Build observability and performance tooling, using production metrics to improve latency, throughput, and resource efficiency.
π οΈ Requirements
- Bachelor's or Master's degree in Computer Science, Engineering, or a related field (or equivalent practical experience)
- 5+ years of experience in model serving, ML infrastructure, or a closely related backend engineering role
- Deep expertise in Python concurrency, including GIL behavior, multi-threading, thread safety, multiprocessing
- Experience building asynchronous and message-driven systems
- High-performance, large-scale distributed systems
- Ability to read and reason about ML model implementations at a computational level, including compute behavior, batching, memory usage, and inference characteristics
- Experience profiling and optimizing performance across CPU, memory, I/O, and ideally GPU workloads, and translating findings into architectural improvements
- Understanding of GPU architecture, performance characteristics, and resource utilization in high-performance compute workloads
- Experience working with tabular and structured-data ML systems
- Understanding of neural networks and modern deep learning architectures
β¨ Benefits
- Competitive compensation with salary and equity
- Comprehensive health coverage for you and your dependents
- Paid parental leave for all new parents, inclusive of adoptive and surrogate journeys
- Relocation support for employees moving to join the team in one of our office locations
- A mission-driven, low-ego culture that values diversity of thought, ownership, and bias toward action
Full job description
About Fundamental
Fundamental is an AI research lab pioneering the future of enterprise decision-making. Our flagship model, NEXUS is the world's most powerful Large Tabular Model (LTM) - purpose-built for the structured records that contain trillions of dollars in business value. With $275m in funding from leading investors and trusted by Fortune 100 companies, Fundamental is giving businesses the Power to Predict..
At Fundamental, you'll work on unprecedented technical challenges in foundation model development and build technology that transforms how the world's largest companies make decisions. This is your opportunity to be part of a category-defining company from the ground-up. Join the team defining the future of enterprise AI.
About the role
Our Serving team is responsible for turning NEXUS, our Large Tabular Model, into a reliable and scalable production system. We own the infrastructure and execution stack that serves the model across multiple deployment environments, each with different requirements around scale, isolation, performance, and trust.
The team sits at the intersection of research and production engineering. We work closely with researchers to bring new model architectures into production, while building the systems needed to operate them efficiently and predictably under real-world workloads. Tabular foundation models introduce serving challenges that differ meaningfully from traditional LLM inference, including irregular computational behavior and complex resource tradeoffs across CPU, GPU, memory, and networking.
As a Model Serving Engineer, youβll work across the full inference stack - from Python runtime performance and concurrency behavior to distributed orchestration, GPU serving infrastructure, and deployment architecture. Youβll identify bottlenecks, improve throughput and latency, and help define how new generations of the model are translated from research artifacts into production-grade systems.
This is a deeply technical, Python-heavy role for engineers who enjoy distributed systems, performance optimization, and low-level infrastructure challenges close to modern ML systems.
Key responsibilities
Optimize Python inference code for performance under real concurrency constraints, including GIL contention, multi-threading, multiprocessing, async execution, and long-running production workloads
Work closely with research to understand model internals and support the continuous evolution of the architecture, especially around complex and non-obvious computational behavior under production load
Collaborate with research and infrastructure teams to reason about hardware utilization and serving tradeoffs across GPU, CPU, memory, networking, batching, and concurrency
Define and evolve the architecture behind our distributed inference and asynchronous execution stack, including orchestration, worker coordination, and end-to-end concurrency patterns
Own the Triton serving layer for NEXUS, including how models are packaged, configured, and executed as part of our production inference pipeline
Build observability and performance tooling across the serving stack, and use production metrics to drive tuning decisions around latency, throughput, and resource efficiency
Solve cross-cutting serving challenges that emerge from deploying the same model across environments with very different scale, isolation, and reliability constraints
Evaluate and integrate new inference runtimes, serving strategies, and infrastructure approaches as the model ecosystem evolves
Must have
Bachelor's or Master's degree in Computer Science, Engineering, or a related field (or equivalent practical experience)
5+ years of experience in model serving, ML infrastructure, or a closely related backend engineering role
Deep expertise in Python concurrency, including GIL behavior, multi-threading, thread safety, multiprocessing
Experience building asynchronous and message-driven systems
High-performance, large-scale distributed systems
Ability to read and reason about ML model implementations at a computational level, including compute behavior, batching, memory usage, and inference characteristics
Experience profiling and optimizing performance across CPU, memory, I/O, and ideally GPU workloads, and translating findings into architectural improvements
Nice to have
Understanding of GPU architecture, performance characteristics, and resource utilization in high-performance compute workloads
Experience working with tabular and structured-data ML systems
Understanding of neural networks and modern deep learning architectures
Experience with Kubernetes and cloud infrastructure
Familiarity with DevOps and production infrastructure tooling, including containers, Helm, observability, and CI/CD systems
Benefits
Competitive compensation with salary and equity
Comprehensive health coverage for you and your dependents
Paid parental leave for all new parents, inclusive of adoptive and surrogate journeys
Relocation support for employees moving to join the team in one of our office locations
A mission-driven, low-ego culture that values diversity of thought, ownership, and bias toward action
Similar jobs
Search more Machine Learning Engineer jobsSenior Machine Learning Engineer, CX
Design and deliver production-scale AI systems for customer support, including multi-agent frameworks, agent copilots, and secure LLM-powered actions that improve resolution and help agents handle cases effectively.
Senior Machine Learning Engineer
Design, build, and operate production optimization applications, developing algorithms and evaluation systems while ensuring reliability and directing AI coding agents to deliver correct, maintainable solutions.
Sr ML Engineer - Remote
Leads the production lifecycle of machine learning systems, building reliable training and inference pipelines, optimizing model performance, and improving monitoring, deployment, and release practices.
Design, deploy, and maintain production-grade AI/ML systems and pipelines, building scalable infrastructure for batch and real-time workloads and bringing models into business applications and downstream processes.
