Staff Machine Learning Engineer, AI Security
Core
Lead the development, training, and optimization of ML models to detect and prevent AI security risks like prompt injection and jailbreaks on Reddit's platform.
Role type
Staff Machine Learning Engineer (AI Security)
Builds
ML-powered security protections for Reddit's AI systems and LLM Guardrails Platform
Domain
Internet / AI Security / Large Language Models
Deliverable
production ML models
Required skills
Python, deep learning frameworks (TensorFlow, PyTorch, Hugging Face), neural network architecture design, scalable data pipelines, distributed training (Ray Train, PyTorch Distributed), experimental design, adversarial testing, model evaluation, data preprocessing, tokenization, embeddings
Preferred skills
ML for security/trust/safety, adversarial training, model distillation, active learning, synthetic-data generation
Technologies
TensorFlow, PyTorch, Hugging Face Transformers, Ray Train, PyTorch Distributed
Responsibilities
Select, adapt, fine-tune, evaluate, and deploy pretrained models and classifiers for security; Build reproducible training and evaluation pipelines; Set technical vision and multi-quarter modeling roadmap; Conduct model evaluations to improve accuracy and adversarial robustness; 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 with technical leadership