Research Engineer, ML Systems (All Industry Levels)
Character.AIover 1 year ago
Foster City, CA, USAEntry Level / Mid Level / Senior
H1B Sponsor
Base Salary
$225k - $400k/yr
Responsibilities
- Optimize GPU performance for AI training and inference systems.
- Write efficient Triton kernels tailored for specific models and hardware.
- Develop algorithms to enhance serving cache hit rates.
- Train and distill large language models to improve latency and engagement.
- Build scalable distributed reinforcement learning from human feedback (RLHF) systems.
Requirements
- PhD or equivalent research experience in a relevant field.
- Proficient in writing clean production system code.
- Strong understanding of modern machine learning techniques, including reinforcement learning and transformers.
- Demonstrated track record of innovative ML systems projects.
- Comfortable with model development code in PyTorch.
Tech Stack
DockerKubernetesPyTorch
Categories
AI & MLData Science