Member of Technical Staff - ML Operations
- Remote, Canada
Remote
Full-time
Applications open in a new tab.
Job details
- Job type
- Full-time
- Workplace
- Remote
- Posting language
- English
- Posted
Original job description
ABOUT US
Veeda AI is building the next generation of multimodal foundation world models for Physical AI. We're a small, fast-moving team of engineers and researchers from leading AI labs, tackling some of the most challenging problems at the intersection of AI, robotics, and embodied intelligence. If you're excited about pushing the boundaries of what's possible with Physical AI, you'll have the opportunity to make an outsized impact from day one.
RESPONSIBILITIES
Experiment Lifecycle Tracking and Tooling: Design, build and deploy tools that own how a run is defined, launched, resumed, and killed. Build and operate experiment databases for code version tracking, data version tracking, reproducibility, and checkpoint ancestry.
Inference Fleet Orchestration: Design, build, and operate the serving control plane that accepts large volumes of concurrent client requests and assigns them across inference clusters. Develop cache-aware admission, routing, batching, and scheduling policies that improve cache locality, balance workload, protect tail latency, and keep the fleet highly utilized and reliable. Partner with ML Performance on model runtime, kernel, and per-worker throughput optimization.
Model Evaluation in CI: Design, build, and operate automatic model checkpoint evaluation systems on seeded rollout and policy-success suites, run per-change and nightly.
Data Pipeline Operations: Design, build, and operate high-performance, fault-tolerant, distributed backend services and event-driven systems for our large scale data processing pipeline.
Visualization Platforms: Design, build, and deploy experiment observability and dataset visualization platform(s) that provides interactive data visualization, progress tracking, search, and comparison.
End-to-End Ownership: Lead projects through the complete software lifecycle, including technical specs, implementation, CI/CD, on-call support, and production observability.
REQUIREMENTS
Bachelor's degree or equivalent hands-on experience in Computer Science, Engineering, or a related technical field.
Proficiency in at least one scripting (Python or Bash) and one compiled (Java, Rust, or Go) languages.
Experience in shipping production-quality developer tools.
Proficient in CICD automations (pipelines, runners, deployment)
Experience in building reproducible pipelines end to end, and can say precisely which parts of a training run are bit-reproducible, which are not, and why.
One of the following:
Proven understanding of event-driven architecture, concurrency models, fault tolerance, and data consistency patterns.
Experience building or operating large-scale inference control planes or distributed serving infrastructure, with hands-on work in traffic management, admission control, request scheduling, routing, batching, or cache-aware load balancing; able to reason clearly about cache locality, queueing, tail latency, availability, and fleet utilization.
Experience working with multi-node workloads and building around slurm based scheduling systems.
Experience working on in-production model evaluation frameworks, in particular regression testing of large multimodal models.
NICE TO HAVE
You have run experiment tracking at scale, logging video, 3D, and trajectory artifacts rather than only scalars.
You have built evaluation harnesses for generative or embodied models, where quality is a distribution rather than a pass/fail.
Full stack development experience with web-based front-end.
You have orchestrated ML workflows with Argo Workflows, Flyte, or Ray, and know where each one breaks.
You have built GPU-hour attribution that maps cluster spend back to specific experiments and teams.
You have contributed to open-source ML tooling, or published on evaluation or reproducibility methodology.
More about this job
Responsibilities
You will own the lifecycle, launch tooling, and experiment tracking for multimodal foundation models. Additionally, you will manage checkpoint registries, evaluation in CI, and incident response for live training runs.
Requirements
You must have a Bachelor's degree in a technical field and strong Python and software engineering skills. Experience with multi-node training jobs, CI/CD pipelines, and rigorous evaluation methodology is required.
Skills
- Python
- CI/CD
- Machine Learning Operations
- PyTorch
- Argo Workflows
- Weights & Biases
- MLflow
- Slurm
- Containerization
- Telemetry
- GPU Infrastructure
- Reproducible Pipelines
- Evaluation Methodology
- Incident Response
- Distributed Systems
Visa sponsorship
Not detected in the job text
Categories
- Technology
- Software
- Engineering
- Data & Analytics
- Science & Research
Keywords
- Machine Learning
- MLOps
- Physical AI
- Robotics
- Embodied Intelligence
- Hydra
- OmegaConf
- Docker
- GitHub Actions
- Buildkite
- Flyte
- Ray
- Prometheus
- Grafana
- OpenTelemetry
- GPU-hour attribution
- Postmortem
- Data Loader
- Checkpointing
- Regression Testing