Machine Learning Infrastructure Engineer
Character.AIover 1 year ago
Foster City, CA, USAMid Level
H1B sponsor
Base Salary
$150k - $350k/yr
Responsibilities
- Provide infrastructure support for ML research and product workloads.
- Build tooling to diagnose cluster issues and hardware failures.
- Monitor deployments, manage experiments, and support ML research operations.
- Maximize GPU allocation and utilization for serving and training.
Requirements
- At least four years of experience supporting infrastructure within a machine learning environment.
- Experience developing tools to diagnose ML infrastructure problems and failures.
- Experience with cloud platforms such as Compute Engine, Kubernetes, and Cloud Storage.
- Experience working with GPUs.
- Experience with large GPU clusters and high-performance computing or networking is preferred.
- Experience supporting large language model training is preferred.
- Experience with PyTorch, TensorFlow, or JAX is preferred.
- Experience with GPU kernel development is preferred.
Tech Stack
Categories
About Character.AI
Character.AI builds a consumer platform for creating and conversing with customizable AI characters on the web and mobile. The company monetizes through a paid c.ai+ subscription and in-app purchases, and reports over 20 million monthly users engaging with user- and creator-built bots. Founded in 2021 by former Google AI leaders and headquartered in Redwood City, CA, it is privately held.