Staff Machine Learning Engineer
Core
Lead the development, training, and optimization of ML models for AI security (prompt injection, jailbreaks, data exposure) on Reddit's LLM Guardrails Platform.
Role type
Staff Machine Learning Engineer (AI Security)
Builds
ML-powered security protections for Reddit's AI systems and products
Domain
Internet / AI Security / Large Language Models
Deliverable
production ML models
Required skills
Python, TensorFlow, PyTorch, Hugging Face Transformers, neural network architecture, distributed training (Ray Train, PyTorch Distributed), experimental design, adversarial testing, model evaluation, data pipeline engineering
Preferred skills
ML for security/trust/safety, adversarial training, model distillation, active learning, synthetic-data generation
Technologies
Ray Train, PyTorch Distributed, Hugging Face Transformers
Responsibilities
Select, adapt, fine-tune, evaluate, and deploy pretrained models and classifiers for security problems; Build reproducible training and evaluation pipelines; Set technical vision and multi-quarter modeling roadmap; Conduct model evaluations and performance analysis; Own training-data quality and production model lifecycle; Establish best practices for responsible ML development; Mentor engineers and lead technical discussions
Seniority
Staff, strategic hands-on IC
