We are seeking a Senior Staff Machine Learning Engineer to join our Model Factory team. The ideal candidate will have strong programming skills in Python and experience with ML frameworks like PyTorch, TensorFlow, or JAX. We value humility and believe in direct communication. Our team is inclusive, and our differing perspectives allow for better solutions.
Requirements
- BS in Computer Science with 7+ years of strong programming skills in Python
- Experience with ML frameworks like PyTorch, TensorFlow, or JAX
- Hands-on experience with model optimization, quantization, and inference acceleration
- Deep understanding of transformer architectures, attention mechanisms, and distributed inference (tensor parallel, pipeline parallel, sequence parallel)
- Knowledge of quantization (INT8, BF16, FP16) and memory-efficient inference techniques
- Solid grasp of software engineering best practices, including CI/CD, containerization (Docker, Kubernetes), and MLOps
Benefits
- Competitive compensation
- Benefits
- Opportunities for career growth
