📅 Mar 13
Build and scale Tower’s machine learning research platform, enabling reproducible experimentation, GPU-accelerated training, simulation, and efficient compute across HPC and cloud environments.
ML Software Engineer
📅 Nov 5, 2025
PythonPytorchTensorflowJAX
📜 Description
- Design and ship multimodal data pipelines and reproducible training and evaluation workflows.
- Implement LLM, VLM, and VLA architectures, including model code, tokenization, inference runners, and output heads.
- Integrate simulators for closed-loop evaluation and build metrics, visualization, and experiment-management tools.
- Deliver production serving and inference tools for deterministic, low-latency operation.
- Own systems end to end, from architecture and implementation through testing, documentation, and iteration.
🛠️ Requirements
- Education & Experience: MS in CS/ML/Robotics or BS + ≥2 years building ML/data/evaluation systems
- Software engineering excellence: Strong Python fundamentals (data structures, testing, debugging, modular design) and a track record of shipping production-quality code/APIs and reliable automation.
- Pipelines → Training → Serving: Demonstrated experience building data/ML pipelines and evaluation tooling, and integrating training and inference using PyTorch, TensorFlow, or JAX.
- Datasets at scale: Dataset packaging, sharding, manifest formats, and integrity checks for large multimodal datasets.
- Performance & optimization: Practical work improving training/inference throughput and latency (e.g., mixed precision, efficient batching, model parallelism).
- MLOps & infrastructure: Cloud storage and training workflows, containerization, CI/CD, and experiment observability (tracking, logging, metrics).
- Team fit: Strong communication, collaborative with research and engineering partners, and a bias for ownership/independence in a small, fast-moving team.
- Nice to have: Prior work on perception, detection, or multimodal models.
Full job description
About us
We’re building the next generation of ground transportation with advanced physical AI to simplify the toughest challenges in modern freight. Our stealth team, founded by the engineers who scaled autonomous driving, is developing an entirely new vehicle platform. We move fast, stay tightly aligned between engineering and product, and focus on creating reliable, real-world autonomous systems.
What You’ll Do
What We’re Looking For
Similar jobs
Search more Machine Learning Engineer jobs
Machine Learning Engineer, Prediction & Planning
📅 Apr 28, 2025
Waymo👥 1001 - 5000 employees🏢 Internet
📅 Apr 28, 2025
Develop machine learning models for autonomous vehicle prediction and planning, turning real-world driving challenges into robust systems that support safe, smooth behavior and integrate into production software.
Machine Learning Performance Engineer (Inference)
📅 Aug 11
Tower Research Capital🏢 Financial Services🤝 B2B
📅 Aug 11
Optimize deep learning inference pipelines for microsecond-level latency, evaluating heterogeneous hardware, developing GPU kernels, and improving model efficiency, reliability, and deployment performance.
Machine Learning Engineer II (ROA + LTV ML)
🕒 7 days ago
Affirm👥 10,000+ employees🏢 Financial Services
🕒 7 days ago
Develop and scale production machine learning models for credit decisioning, including default prediction and approval decisions, while building training, monitoring, and data systems with engineering, credit, and product teams.
🕒 9 days ago
Build and evolve scalable AI/ML infrastructure that helps product teams develop, deploy, and operate AI-powered applications, including model training and serving, data workflows, orchestration, and evaluation tools.
📍 🇺🇸 United States - Remote💰 $145.4k - $188.1k / year💼 Full-Time🥁 Junior🗣️ Machine Learning Engineer