Research Engineer, Infrastructure
Core
Build the systems that power how the company adapts and improves leading open-weight models for legal work, including distributed training, RL rollout systems, and agent environments.
Role type
Senior IC research engineer (ML infrastructure)
Builds
Distributed training and RL rollout infrastructure, reproducible agent environments, internal tools for data management and experiment tracking, and production serving systems.
Domain
Legal technology + Machine Learning infrastructure
Deliverable
production ML models
Required skills
ML infrastructure, distributed systems, Python, PyTorch/JAX, debugging across stack, performance optimization, reliability engineering, observability, data versioning, model serving
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, identify bottlenecks, build internal tools for data and experiments, integrate models into production, partner with researchers to operationalize workflows
Seniority
Senior, hands-on IC
