Software Engineer - Training Product

Job summary

Responsibilities

We’re looking for a customer-obsessed software engineer to come ship with us. You’ll own features like multi-node training and products like serverless reinforcement learning (RL) from conception to MVP (and from MVP to GA!).

• Iterate like crazy
• Design ergonomic APIs and abstractions to model complex resources and lifecycles
• Work throughout the stack (API layer, backend and database implementation, infra layer; frontend is a plus) to implement features.
• Fine-tune and deploy models to develop intuition around training workflows.
• Partner closely with model developers and world-class research engineers to understand the requirements and pain points of post-training workflows.

What they're looking for

• 5+ years experience building software applications
• Deep knowledge of the web stack, databases, and distributed systems
• Experience developing developer tooling or infrastructure products for external or internal users.
• Good taste in product, particularly developer-oriented tools
• Interest in ML/AI infrastructure and willingness to learn

Tech stack

• Spark
• vLLM
• Kubernetes
• PyTorch