Jobs › Companies › Rockstar › Backend Software Engineer (ML Infra)

About this Backend Software Engineer (ML Infra) role at Rockstar

Rockstar · Onsite · San Francisco, California, United States

Rockstar is recruiting for a fast-growing AI infrastructure startup building the optimal inference allocation engine—model routing, dedicated inference, and post-training, all working together to ensure every request is served by the best model for it, in real time. The client’s mission is to deliver truly serverless intelligence to the AI world. Their routers and models allocate and serve 100B+ tokens/month for frontier AI companies across industries.

About the Client

The client is building the optimal inference allocation engine—model routing, dedicated inference, and post-training, all working together to ensure every request is served by the best model for it, in real time. Their mission is to deliver truly serverless intelligence to the AI world.

Their routers and models allocate and serve 100B+ tokens/month for frontier AI companies across industries.

The Role

The client is hiring a Software Engineer to help design, build, and scale the core systems that power large-scale routing, model training, and deployment.

The engineer will work on everything from distributed training pipelines, cloud-native infrastructure, and internal developer platforms that support fine-tuning, reinforcement learning, and inference at scale, stretching right up to the frontend. This role sits at the intersection of product engineering and ML systems—the engineer will collaborate closely with ML scientists while owning production-grade infrastructure operating at 99.99% uptime, alongside a team that has shipped hundreds of models at places like TikTok, MongoDB, and Neeva.

This is an ideal role for an early-career engineer who wants to work on real distributed systems, GPU workloads, and modern ML infrastructure, and actually see their work transform the AI capabilities of frontier application layer companies.

What You’ll Do

Build & Scale Core Infrastructure

Design and implement systems that support large-scale ML workloads, including fine-tuning and reinforcement learning.

Build distributed training, inference, and logging pipelines that are efficient, fault-tolerant, and observable.

Cloud & Systems Engineering

Work on cloud-native systems using containers and orchestration (e.g., Kubernetes).

Optimize systems for performance, reliability, and cost efficiency, especially for GPU-heavy workloads.

Implement monitoring, logging, and observability for long-running training jobs and production services.

Bring the Client’s Capabilities to Customers

Talk to customers. Join their calls, find what’s slowing them down, and ship the fix yourself.

Realize the client’s features. Develop the frontend and backend logic that enables customers to access the client’s capabilities.

Who You Are

Required

1–3 years of engineering experience, ideally working on production systems on the backend and infrastructure layer.

Strong fundamentals in distributed systems, networking, and backend architecture.

Comfortable writing code with and without AI and taking accountability for your code.

Excited to work on-site in San Francisco with a fast-moving early-stage team.

Ready to apply to Rockstar?
Apply to Rockstar

Similar jobs

Redwood Materials
Software Engineer - ML/Computer Vision (Battery Sorting)
Redwood Materials
⚡ Apply early McCarran, NV; San Francisco, C... Onsite $152,500–$200,000
● New 👁 Seen ✓ Applied 4h ago
Redwood Materials
Software Validation Engineer, Energy Storage
Redwood Materials
⚡ Apply early San Francisco, California, Uni... Onsite $152,500–$342,500
● New 👁 Seen ✓ Applied 5h ago
Redwood Materials
Embedded Software Engineer – Power Electronics, Energy Storage
Redwood Materials
⚡ Apply early San Francisco, California, Uni... Onsite $180,000–$237,500
● New 👁 Seen ✓ Applied 5h ago
Flexport
Staff Software Engineer, Autonomous Freight Systems
Flexport
⚡ Apply early San Francisco, California, Uni... Onsite $196,875–$270,703
● New 👁 Seen ✓ Applied 6h ago
Flexport
Senior Software Engineer, Autonomous Freight Systems
Flexport
⚡ Apply early San Francisco, California, Uni... Onsite $206,181–$252,000
● New 👁 Seen ✓ Applied 6h ago
Taskrabbit
Staff Software Engineer - Experimentation Platform
Taskrabbit
⚡ Apply early San Francisco, California, Uni... Hybrid $160,000–$215,000
● New 👁 Seen ✓ Applied 23h ago
Taskrabbit
Staff Software Engineer - Authentication
Taskrabbit
⚡ Apply early San Francisco, California, Uni... Hybrid $160,000–$215,000
● New 👁 Seen ✓ Applied 23h ago
Checkr
Senior Software Engineer (Ruby/Rails), MVR
Checkr
⚡ Apply early San Francisco, California, Uni... Onsite $185,000–$218,000
● New 👁 Seen ✓ Applied 23h ago
a16z
Software Engineer, Systems (Partner 16, Partner 18)
a16z
⚡ Apply early San Francisco, California, Uni... Hybrid $301,000–$352,000
● New 👁 Seen ✓ Applied 1d ago

Sign up for suggestions tailored to the jobs you open and the searches you save.

More jobs at Rockstar

See all jobs at Rockstar →

Apply now
🤖

Whoa — hold up

JobsRadar was built for real people having a rough time in their job search — not for automated requests. You're clicking way too fast and you're now temporarily blocked.

Come back later. If you're genuinely job hunting, we've got your back — just act like a human.

Catch your next role the second it’s posted.

Create a free account and we’ll watch the boards for you — the instant a job matches your search, it lands in your inbox or Telegram. No digging, no refreshing.

Create free account

Free forever · takes 30 seconds · already have one?

Get an edge on your job hunt.

Join our Telegram channel for the stuff that helps you land the role — salary benchmarks, the weekly market pulse, and new-feature drops. No spam, just signal.

Join the channel — it's free