Build, remix and share AI-powered educational tools.
Staff Machine Learning Engineer
Location
United States
Posted
94 days ago
Salary
$190K - $250K / year
Seniority
Lead
Job Description
Staff Machine Learning Engineer
Playlab
• Design and build evaluation systems that assess educational AI quality across thousands of conversations - from learning outcomes to bias detection to curriculum alignment • Build ML systems that enable self-improving app creation - learning from high-quality apps on the platform to automatically scaffold new applications for educators • Design and prototype downloadable, on-device AI models that work without internet connectivity - critical for privacy and global accessibility • Develop systems that enable dynamic, fluid interfaces adapting to learning moments - transitioning seamlessly from chat to writing editor to interactive physics simulation as needed • Build content moderation and safety systems designed specifically for educational discourse • Implement agentic AI systems that enable educators to create goal-directed applications (e.g., "help students through this project over 2 weeks") • Build sophisticated RAG systems that integrate diverse educational content with semantic search and knowledge graphs
Job Requirements
- 7+ years building and deploying ML systems in production, with recent experience in generative AI and LLMs
- Strong understanding of ML fundamentals, model fine-tuning, and evaluation methodologies
- Experience building production AI systems - you understand latency, cost optimization, and evaluation challenges
- Proficient in Python and ML frameworks (PyTorch, TensorFlow, HuggingFace, etc.)
- Thrive in high-agency, high collaboration cultures
Benefits
- Professional development
- Remote work options
Related Guides
Related Job Pages
More Machine Learning Engineer Jobs
• Token Factory is a part of Nebius Cloud, one of the world's largest GPU clouds, running tens of thousands of GPUs. • We are building a high-performance inference and fine-tuning platform designed to push foundation models to their hardware limits. • Our mission is to maximize throughput, minimise latency, and optimise cost-per-token across tens of thousands of GPUs. • Inference Optimization: Identifying LLM inference bottlenecks to drive production speedups. • Inference engines support: Implement novel speculative decoding architectures, optimise components of various LLM designs (dense/MoE, autoregressive/parallel), and contribute to open-source inference engines. • Low Precision Training & Inference: Design and productionise low-precision (FP8, NVFP4/MXFP4) training and inference pipelines with measurable gains in throughput and cost-efficiency.
• Ship and support AI-native products from concept to production. • Apply sound judgment on building and maintaining high-performing production ML systems. • Stay deeply informed on the latest machine learning research and developments. • Collaborate with a team of strong engineers; serve as a mentor and resource for other high-potential ML engineers.
Senior Machine Learning Engineer
PartnerOneWe are the leaders in Big Data management through hyper-automation, virtualized cloud tiering, metadata and AI
• Implement and integrate agent-based systems into operational workflows. • Build, deploy, and monitor ML/AI models in production (batch). • Design, build, and maintain large-scale geospatial data pipelines. • Develop backend services and ML tooling • Establish observability for pipelines, models, and agents (metrics, tracing, alerting). • Collaborate with product and customer teams to drive revenue.
• Design and implement deep learning models for 3D computer vision tasks, including object detection, segmentation, and depth estimation. • Develop and maintain end-to-end machine learning pipelines encompassing data preprocessing, model training, evaluation, and deployment. • Optimize models for real-time inference and deploy them using cloud platforms such as AWS SageMaker or GCP Vertex AI. • Monitor deployed models, analyze performance metrics, and implement retraining strategies to ensure sustained accuracy and reliability. • Document methodologies, experiments, and findings; actively participate in code reviews and technical discussions. • Stay abreast of the latest research and advancements in machine learning and computer vision to inform model development.
