Research Engineer, Infrastructure
Core
Build the infrastructure for scaling post-training (SFT and RL) of open-weight models for legal work, including distributed training, rollout systems, and agent environments.
Role type
Senior IC research engineer (ML infrastructure)
Builds
Distributed training systems, RL rollout infrastructure, reproducible agent environments, and internal tools for data management and experiment tracking.
Domain
Legal technology + Machine Learning Infrastructure
Deliverable
production ML models
Required skills
Distributed systems, ML infrastructure, Python, PyTorch/JAX, performance optimization, reliability engineering, observability
Preferred skills
RL rollouts, agent environments, sandboxed tool execution, post-training workflows
Technologies
PyTorch, JAX, Python
Responsibilities
Scale post-training stack for SFT and RL; build distributed rollout infrastructure; create reproducible agent environments; own performance and reliability of training runs; remove bottlenecks in training and inference; build internal tools for data management and experiment tracking; integrate models into production.
Seniority
Senior, hands-on IC
