remote
Machine Learning Engineer III, Core Agents - Box
ML Engineer
Senior machine learning engineer building core AI agents, responsible for model design, training, deployment, and scaling on cloud infrastructure using Python, TensorFlow/PyTorch, and container orchestration.
About the role
Key Responsibilities
- Design, develop, and optimize machine learning models for core AI agents that power content management workflows.
- Implement end‑to‑end pipelines for data ingestion, preprocessing, training, evaluation, and continuous monitoring.
- Deploy and scale models on cloud platforms using AWS services, Docker containers, and Kubernetes orchestration.
- Collaborate with product, security, and infrastructure teams to integrate AI capabilities into the platform while ensuring performance, reliability, and compliance.
- Research and apply state‑of‑the‑art techniques in natural language processing, computer vision, and reinforcement learning to improve agent capabilities.
Requirements
- 5+ years of professional experience in machine learning engineering, with a strong focus on model development and productionization.
- Proficiency in Python and deep learning frameworks such as TensorFlow or PyTorch.
- Hands‑on experience with cloud services (AWS), containerization (Docker), and orchestration (Kubernetes) for scalable deployments.
- Solid understanding of data pipelines, feature engineering, and model evaluation metrics.
- Demonstrated ability to work cross‑functionally, communicate complex concepts clearly, and deliver production‑ready AI solutions.
Skills
pythontensorflowpytorchawskubernetesdocker