Why this role stands out
This role offers a unique opportunity to shape the future of AI inference by building cutting-edge systems for novel hardware, presenting significant growth and learning potential. You will thrive here if you are passionate about solving complex technical challenges in distributed systems and ML infrastructure, and you will be instrumental in optimizing performance for leading AI companies. Apply now to be at the forefront of AI innovation and contribute to groundbreaking advancements in the field.
Quick overview
Job description
About us
Gimlet is building the first multi-silicon neocloud designed for fast, efficient AI inference.
We combine large-scale compute infrastructure with an execution platform that partitions AI workloads and maps each stage to the hardware best suited to run it.
We work with foundation labs, hyperscalers, and AI-native companies, giving our team access to technical problems spanning frontier models, production infrastructure, and emerging hardware.
About the role
As a Member of Technical Staff focused on ML Systems, you will build the inference systems that execute models end-to-end in production.
You will work on the systems that determine how inference executes across that pipeline: how requests are batched and scheduled, how stages are placed and scaled, how KV cache and intermediate state move between accelerators, and how the system balances latency, throughput, and utilization across different hardware characteristics.
You will work across model serving, batching, scheduling, concurrency, KV cache management, and memory placement. You will help bring up models on novel hardware. You will support new model architectures and inference techniques, improve performance under real production workloads, and partner with compiler, kernel, networking, and distributed systems engineers to optimize the full execution path.
What success looks like
In the first 12-18 months, you will:
Improve the latency, throughput, and efficiency of production inference workloads
Design execution strategies across batching, scheduling, concurrency, and resource utilization
Improve KV cache management, memory efficiency, and execution under load
Enable new models, accelerator architectures, and inference techniques to run efficiently in production
You may be a good fit if you have
Strong software engineering fundamentals
Experience building or operating ML inference or model serving systems
Comfort reasoning about performance, memory usage, and system behavior under load
Bachelor's degree in a relevant field, or an equivalent combination of education, training, and professional experience.
Strong candidates may also have
Experience with inference runtimes such as TensorRT-LLM, vLLM, or custom serving systems
Deep understanding of modern model architectures and attention mechanisms
Experience with batching, scheduling, and concurrency control in inference systems
Familiarity with KV cache management and memory placement strategies
Experience profiling and tuning latency- and throughput-critical systems
Software development experience in Python and C++
Why join now?
Gimlet is expanding from its core technology into a production neocloud spanning new hardware, customers, and data centers.
Solve hard problems.
Own meaningful work.
Build for production.
Help define what’s next.
Agency Policy: Gimlet Labs does not accept unsolicited resumes from recruitment agencies or search firms. Any unsolicited resumes submitted without a signed agreement will be considered the property of Gimlet Labs, and no fees will be paid.
Similar jobs
- SE
Senior Staff ML Engineer, AI Safeguards - MoveworksRecently posted Auto Apply
ServiceNow · Mountain View, CALIFORNIA · On-site 🇺🇸
LLMPythonServiceNowTechnologySalary not listed - SC
Machine Learning Infrastructure Engineer - LLM & Agentic AIRecently posted Auto Apply
Stellar Cyber · United States · Remote 🇺🇸
DockerMicroservicesMLOps+12TechnologySalary not listed - SE
Senior Machine Learning Engineer - Active LearningRecently posted Auto Apply
Sense · Cambridge, Massachusetts · Hybrid 🇺🇸
SQLDeep LearningPython+2TechnologySalary not listed - NO
Intelligent Automation/AI-ML EngineerRecently posted Auto Apply
Node.Digital · Herndon, Virginia · Hybrid 🇺🇸
DockerGCPAWS+21TechnologySalary not listed - FP
Scientist, Machine Learning (Human Physiology)Recently posted Auto Apply
Flagship Pioneering, Inc. · Cambridge · Hybrid 🇺🇸
Applicant Tracking SystemsGDPRGreenhouse+3HealthcareSalary not listed - CL

Founding ML Engineer, Computer Vision (Object Detection)Recently posted Auto Apply
Clera · New York · Remote 🇺🇸
Computer VisionPyTorchTensorFlow+1TechnologySalary not listed - ZO

Machine Learning Engineer - Vision Foundation ModelsRecently posted Auto Apply
Zoox · Foster City · Hybrid 🇺🇸
Technology$254k - $305k/yr - MI

Machine Learning Engineer - Pre Training Auto Apply
Mindbeam · United States · Remote 🇺🇸
DockerPyTorchPython+4TechnologySalary not listed - SM

Cleared AI/ML Engineer (5785)Recently posted Auto Apply
SMX · United States · Hybrid 🇺🇸
DatabricksLLMMachine Learning+1TechnologySalary not listed - RA
Member of Technical Staff - Applied MLRecently posted Auto Apply
Raydar · New York · On-site 🇺🇸
LLMPythonMachine LearningSalary not listed - SE
Machine Learning Engineer Auto Apply
ServiceNow · Santa Clara, CALIFORNIA · Hybrid 🇺🇸
JavaScriptJavaDocker+3TechnologySalary not listed - SE
Principal Machine Learning Engineer Auto Apply
ServiceNow · Santa Clara, California · Hybrid 🇺🇸
Generative AITechnologySalary not listed
