about 4 hours ago
Palo Alto, CA, USASenior / Mid Level
H1B Sponsor
Base Salary
$189k - $283k/yr
Responsibilities
- Own the full training lifecycle for small language models and classifiers in SAGE's real-time enforcement path.
- Design and implement adversarial training pipelines to improve model robustness.
- Engineer high-performance model serving and inference infrastructure for real-time and batch workloads.
- Build automated data curation and evaluation frameworks to enhance model training.
- Mine insights from agent interactions to propose new policies and improve existing ones.
- Collaborate with cross-functional teams to align modeling efforts with customer governance requirements.
Requirements
- Bachelor's degree in Computer Science, Machine Learning, or a related field.
- 2+ years of professional ML experience with end-to-end production ownership.
- Proficiency in Python and PyTorch for model training and evaluation.
- Hands-on experience with training and fine-tuning language models in production.
- Experience with serving frameworks and optimization techniques for inference.
- Comfort operating at production scale, especially in safety-critical environments.