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 isn’t just another AI chat app—it’s an infinite playground for your imagination, creativity, and exploration. Dive into immersive conversations, create your own AI chatbots, and join a vibrant community of creators.