Machine Learning Infrastructure Engineer

Job Overview

We’re looking for seasoned ML Infrastructure engineers with experience designing, building and maintaining training and serving infrastructure for ML research. This is a remote work opportunity for a key engineering position.

Key Responsibilities

• Provide infrastructure support to our ML research and product


• Build tooling to diagnose cluster issues and hardware failures


• Monitor deployments, manage experiments, and generally support our research


• Maximize GPU allocation and utilization for both serving and training

Required Qualifications

• 4+ years of experience supporting the infrastructure within an ML environment


• Experience in developing tools used to diagnose ML infrastructure problems and failures


• Experience with cloud platforms (e.g., Compute Engine, Kubernetes, Cloud Storage)


• Experience working with GPUs

Nice to Have

• Experience with large GPU clusters and high-performance computing/networking


• Experience with supporting large language model training


• Experience with ML frameworks like Pytorch/TensorFlow/JAX


• Experience with GPU kernel development

About Character.AI

Character.AI empowers people to connect, learn and tell stories through interactive entertainment. Over 20 million people visit Character.AI every month, using our technology to supercharge their creativity and imagination. Our platform lets users engage with tens of millions of characters, enjoy unlimited conversations, and embark on infinite adventures. In just two years, we achieved unicorn status and were honored as Google Play's AI App of the Year—a testament to our innovative technology and visionary approach.

Benefits

• Top-notch health coverage for you & your family, with majority of the premium covered


• We invest in your future with a generous 401(K) contribution


• New parents, we've got you covered with incredible paid leave - up to 20 weeks


• 4 weeks of PTO to explore, unwind & come back recharged


• Daily in-office catering plus a monthly Doordash stipend to help keep you fueled no matter where you are


• Monthly wellness stipend to support you in your health journey