About the job
As a Research Engineer on our team you will work end to end across the whole model stack, identifying and addressing key infra blockers on the path to scientific AGI. Strong candidates should have familiarity with elements of language model training, evaluation, and inference and eagerness to quickly dive and get up to speed in areas they are not yet an expert on. This may include performance optimization, distributed systems, VM/sandboxing/container deployment, and large scale data pipelines.
Responsibilities
Design and implement large-scale infrastructure systems to support AI scientist training, evaluation, and deployment across distributed environments
Identify and resolve infrastructure bottlenecks impeding progress toward scientific capabilities
Develop robust and reliable evaluation frameworks for measuring progress towards scientific AGI.
Build scalable and performant VM/sandboxing/container architectures to safely execute long-horizon AI tasks and scientific workflows
Collaborate to translate experimental requirements into production-ready infrastructure
Develop large scale data pipelines to handle advanced language model training requirements
Optimize large scale training and inference pipelines for stable and efficient reinforcement learning
Qualifications
Minimum
Have 6+ years of highly-relevant experience in infrastructure engineering with demonstrated expertise in large-scale distributed systems
Are a strong communicator and enjoy working collaboratively
Possess deep knowledge of performance optimization techniques and system architectures for high-throughput ML workloads
Have experience with containerization technologies (Docker, Kubernetes) and orchestration at scale
Have proven track record of building large-scale data pipelines and distributed storage systems
Excel at diagnosing and resolving complex infrastructure challenges in production environments
Can work effectively across the full ML stack from data pipelines to performance optimization
Have experience collaborating with other researchers to scale experimental ideas
Thrive in fast-paced environments and can rapidly iterate from experimentation to production
Preferred
Experience with language model training infrastructure and distributed ML frameworks (PyTorch, JAX, etc.)
Background in building infrastructure for AI research labs or large-scale ML organizations
Knowledge of GPU/TPU architectures and language model inference optimization
Experience with cloud platforms (AWS, GCP) at enterprise scale
Familiarity with VM and container orchestration.
Experience with workflow orchestration tools and experiment management systems
History working with large scale reinforcement learning
Comfort with large scale data pipelines (Beam, Spark, Dask, …)