Key skills
PythonPyTorchAIMachine LearningMLJAX
About this role
Role Overview
- Build core ML systems for a proactive AI product
- Own end-to-end ML pipelines: data preparation, training, evaluation, inference
- Collaborate with research, product, and engineering teams
- Mentor and review work from other ML engineers
Requirements
- 5+ years of experience in machine learning
- Proficiency in Python, PyTorch / JAX
- Strong understanding of ML models and their behavior in production
- Ability to debug model failures and system issues
- Experience with GPU-based training and inference systems
Tech Stack
- Python
- PyTorch
Benefits
- Health insurance
- Professional development opportunities