Tri Source International

Tri Source International

Senior Research Engineer, Computer Vision (LFV/WFM)

Los Altos, CA · Senior

Sponsorship not specifiedDetected 22 days ago
PythonNode.jsAWSDockerKubernetesLinuxMachine LearningDeep LearningPyTorchData EngineeringNLPComputer VisionRoboticsElectrical EngineeringResearchCollaboration

About the role

  • As our WFMs scale in both capability and ambition, we need a strong engineer who can bridge the gap between research ideas and production-grade systems.
  • You will work closely with researchers to prototype new ideas, run experiments, and help ship our most successful models toward real-world applications with real-world impact.

Responsibilities

  • Collaborate directly with research scientists to implement, iterate on, and evaluate new architectures, objectives, datasets, and training strategies.
  • Build and maintain scalable pipelines for ingesting, converting, validating, and serving heterogeneous datasets (multi-view, multi-modal, multi-embodiment, etc.), across robotics and autonomous driving, into unified training-ready formats.
  • Support and optimize large-scale distributed training of world foundation models on multi-GPU and multi-node clusters. Manage experiment workflows, profiling, debugging, and hyperparameter sweeps to ensure optimal performance in a timely manner.
  • Develop tools for dataset inspection, experiment tracking, model evaluation, GPU resource management, and visualization. Automate repetitive workflows to improve team velocity.
  • Work with other TRI teams and Toyota affiliates to set up shared pipelines, onboard their data, and support joint training and evaluation efforts.
  • Support and optimize large-scale distributed training of world foundation models on multi-GPU and multi-node clusters.
  • Manage experiment workflows, profiling, debugging, and hyperparameter sweeps to ensure optimal performance in a timely manner.
  • Develop tools for dataset inspection, experiment tracking, model evaluation, GPU resource management, and visualization.

Requirements

  • Deep proficiency in Python, PyTorch, and the Unix/Linux toolchain.
  • Comfort working in terminal-heavy, SSH-based workflows on shared GPU clusters.
  • Experience with video diffusion models, 3D/4D reconstruction, and multi-view geometry.
  • You are proactive, self-directed, and comfortable operating with ambiguity in a research-driven environment that spans multiple divisions.
  • You are a reliable teammate who communicates clearly and takes ownership of problems end-to-end.
  • Master's or PhD in Computer Science, Electrical Engineering, Machine Learning, or a related field, with a minimum of 3 years of relevant experience and strong software engineering skills.
  • Deep proficiency in Python, PyTorch, and the Unix/Linux toolchain. Comfort working in terminal-heavy, SSH-based workflows on shared GPU clusters.
  • Hands-on experience with large-scale deep learning training, including distributed training (DDP, FSDP, DeepSpeed, or similar), GPU profiling, and debugging training failures at scale.
  • Experience building data pipelines for heterogeneous or multi-modal datasets (images, video, depth, point clouds, actions, etc).
  • Bonus Qualifications

Nice to have

  • Experience with cloud training infrastructure (AWS SageMaker, EC2) and containerized workflows (Docker, Kubernetes).
  • Familiarity with standard data formats and collection pipelines (ROS, MCAP, HDF5, etc.) as well as simulation environments.
  • Proficiency with modern AI-assisted development tools (e.g., Copilot, Cursor, Claude Code) for accelerating engineering workflows.
  • Track record of contributions to open-source projects or publications at top venues (CVPR, ICLR, NeurIPS, RSS, ICRA, etc.) is a plus but not required.
  • Please include links to any relevant open-source contributions or technical project write-ups with your application.
  • It is unlawful in Massachusetts to require or administer a lie detector test as a condition of employment or continued employment.
  • An employer who violates this law shall be subject to criminal penalties and civil liability.
  • Pursuant to the San Francisco Fair Chance Ordinance, we will consider qualified applicants with arrest and conviction records for employment.

Compensation

  • The pay range for this position at commencement of employment is expected to be between $180,000 and $258,750/year for California-based roles.
Apply directly at Tri Source International →Create a free account for alerts like thisView Tri Source International immigration profile

This listing is sourced directly from Tri Source International's careers page and normalized into a canonical job model.