joblet.ai
Find JobsNearby JobsJobs for you
Sign inEmployers / Post a Job
joblet.ai

AI-powered job search connecting talent with opportunity.

ELEVEN AI, Inc.
200 Continental Drive, Suite 401
Newark, DE 19713

Product

  • Browse Jobs
  • Job Locations
  • Browse by Companies
  • Post a Job
  • Blog
  • FAQ
  • Jobs Near Me

Company

  • About Us
  • Contact
  • Refer & Earn
  • Explore all pages

Legal

  • Privacy Policy
  • Cookie Policy
  • Terms of Service

Browse jobs by industry

  • AI
  • IT Services
  • Healthcare
  • Manufacturing & Production
  • Supply Chain
  • Infrastructure
  • Transport & Logistics
  • Real Estate
  • Finance & Accounting
  • Consulting
  • Sales & Marketing
  • Hospitality
  • Media & Entertainment
  • Education

© 2026 ELEVEN AI, Inc. joblet.ai is a product of ELEVEN AI, Inc. All rights reserved.

Overview

Company
BaseTen
Location
all cities, TX 44
Employment type
On-site
  • Supplier Quality Engineer - Penang (44)
  • Research Assistant - College of Engineering - Department of Chemical Engineering (44)
  • Principal Data Center Electrical Engineer (44)
  • Director, Strategic Development & Advocacy Ophthalmology Midwest (44)
  • Senior Director of Engineering (44)
  • Distinguished Architect - Security (44)
Back to Jobs
B
BaseTenVerified Employer

Business Services & Consulting • all cities, TX 44

Post-Training Research Engineer (44)

all cities, TX 44On-sitePosted 12 hours ago
Business Services & Consulting

About the Role

Baseten Engineer Position

Baseten powers mission-critical inference for the world's most dynamic AI companies. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $300M Series E, backed by investors including BOND, IVP, Spark Capital, Greylock, and Conviction. Join us and help build the platform engineers turn to to ship AI products.

We are looking for an engineer with strong experience in machine learning and solid foundations in maths and computer science to join our growing Post-Training team at Baseten.

Custom models are instrumental to the success of Baseten customers. By inference volume, the overwhelming majority of traffic at Baseten is to and from models that have been post-trained in some way. The Post-Training team is responsible for the success of our customers' post-trained models, and we employ a wide array of techniques to produce models that are more efficient and higher quality than even the biggest closed source models for the customer's specific needs.

Your role as a research engineer is to build the in-house tooling to support all of this. We care about training a wide spectrum of different model architectures with a variety of techniques efficiently and at scale. At times this involves zooming deep into a particular technical topic, but more often it involves working across the stack as a whole - systems-level concepts like Kubernetes, cgroups, storage systems, and networking topologies, as well as PyTorch distributed tensor computation, and GPU kernels.

Recent Research
  • Dense, on-policy or both?
  • Repeated kv cache for long-running agents
  • Distillation without the dark – replicating black-box on-policy distillation on Baseten

We don't have a rigid set of skills, but here's some of what we're looking for:

  • A deep understanding of modern ML techniques and tools for training transformers
  • Advanced experience in a tensor/array computation library like PyTorch, TensorFlow, Jax, or similar
  • A detailed understanding of transformer training parallelism strategies like data parallelism, sharded data parallelism, tensor parallelism, pipeline parallelism, context parallelism
  • The experience and knowledge to profile and improve the performance of a distributed GPU program in PyTorch or a similar library
  • The ability to perform roofline analysis on a transformer training setup
  • A willingness to dive into messy problems, work with researchers, derive specifications by asking important questions, and execute
  • Familiarity with HPC and distributed computing platforms like Slurm, Ray, Kubernetes, and Dask
  • Familiarity with cluster networking technology like Infiniband, RoCE, GPUDirect
  • Solid fundamentals in operating systems concepts like processes, files, kernel drivers, containerisation, and networking protocols
  • A sense of creativity and willingness to ask difficult questions about our approach, assumptions, and tooling choices
Benefits
  • Competitive compensation, including meaningful equity.
  • 100% coverage of medical, dental, and vision insurance for employee and dependents
  • Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
  • Paid parental leave
  • Fertility and family-building stipend through Carrot
  • Company-facilitated 401(k)
  • Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

Apply now to embark on a rewarding journey in shaping the future of AI! If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you.

At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status.

We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law.

Baseten Engineer Position

Baseten powers mission-critical inference for the world's most dynamic AI companies. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $300M Series E, backed by investors including BOND, IVP, Spark Capital, Greylock, and Conviction. Join us and help build the platform engineers turn to to ship AI products.

We are looking for an engineer with strong experience in machine learning and solid foundations in maths and computer science to join our growing Post-Training team at Baseten.

Custom models are instrumental to the success of Baseten customers. By inference volume, the overwhelming majority of traffic at Baseten is to and from models that have been post-trained in some way. The Post-Training team is responsible for the success of our customers' post-trained models, and we employ a wide array of techniques to produce models that are more efficient and higher quality than even the biggest closed source models for the customer's specific needs.

Your role as a research engineer is to build the in-house tooling to support all of this. We care about training a wide spectrum of different model architectures with a variety of techniques efficiently and at scale. At times this involves zooming deep into a particular technical topic, but more often it involves working across the stack as a whole - systems-level concepts like Kubernetes, cgroups, storage systems, and networking topologies, as well as PyTorch distributed tensor computation, and GPU kernels.

Recent Research
  • Dense, on-policy or both?
  • Repeated kv cache for long-running agents
  • Distillation without the dark – replicating black-box on-policy distillation on Baseten

We don't have a rigid set of skills, but here's some of what we're looking for:

  • A deep understanding of modern ML techniques and tools for training transformers
  • Advanced experience in a tensor/array computation library like PyTorch, TensorFlow, Jax, or similar
  • A detailed understanding of transformer training parallelism strategies like data parallelism, sharded data parallelism, tensor parallelism, pipeline parallelism, context parallelism
  • The experience and knowledge to profile and improve the performance of a distributed GPU program in PyTorch or a similar library
  • The ability to perform roofline analysis on a transformer training setup
  • A willingness to dive into messy problems, work with researchers, derive specifications by asking important questions, and execute
  • Familiarity with HPC and distributed computing platforms like Slurm, Ray, Kubernetes, and Dask
  • Familiarity with cluster networking technology like Infiniband, RoCE, GPUDirect
  • Solid fundamentals in operating systems concepts like processes, files, kernel drivers, containerisation, and networking protocols
  • A sense of creativity and willingness to ask difficult questions about our approach, assumptions, and tooling choices
Benefits
  • Competitive compensation, including meaningful equity.
  • 100% coverage of medical, dental, and vision insurance for employee and dependents
  • Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
  • Paid parental leave
  • Fertility and family-building stipend through Carrot
  • Company-facilitated 401(k)
  • Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

Apply now to embark on a rewarding journey in shaping the future of AI! If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you.

At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status.

We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law.

What You'll Do

Dense, on-policy or both?
Repeated kv cache for long-running agents
Distillation without the dark – replicating black-box on-policy distillation on Baseten
A deep understanding of modern ML techniques and tools for training transformers
Advanced experience in a tensor/array computation library like PyTorch, TensorFlow, Jax, or similar
A detailed understanding of transformer training parallelism strategies like data parallelism, sharded data parallelism, tensor parallelism, pipeline parallelism, context parallelism

Skills & Technologies

Business Services & Consulting

Similar jobs

Supplier Quality Engineer - Penang (44)
Zimmer Biomet
all cities, TX 44Posted 9 days ago
Research Assistant - College of Engineering - Department of Chemical Engineering (44)
Carnegie Mellon University
all cities, TX 44Posted 3 days ago
Principal Data Center Electrical Engineer (44)
EdgeConneX
all cities, TX 44Posted 3 days ago
Director, Strategic Development & Advocacy Ophthalmology Midwest (44)
ANI Pharmaceuticals
all cities, TX 44Posted 8 days ago
Senior Director of Engineering (44)
Computrition
all cities, TX 44Posted 5 days ago
Distinguished Architect - Security (44)
Oak St. Health
all cities, TX 44Posted 3 days ago
B
BaseTen
Business Services & Consulting
View all jobs at BaseTen