Research Engineer, Discovery

Job summary

Responsibilities

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…

• 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

Tech stack

• Docker
• Kubernetes
• PyTorch
• JAX
• AWS
• GCP
• Spark