Character

Machine Learning Infrastructure Engineer

Redwood City, CA, Remote. Posted 18 months ago

The role

This person will design, build and maintain training and serving infrastructure for ML research. They will provide infrastructure support to ML research and product.

What you will do

  • 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 the company

Character.AI empowers people to connect, learn and tell stories through interactive entertainment. The platform lets users engage with tens of millions of characters.