Staff ML Engineer, VLA Training

The posting, as Nidus Technologies wrote it

ABOUT US

Nidus is building autonomous manufacturing systems powered by AI and robotics. U.S. manufacturers are under growing pressure to increase output while navigating workforce constraints, underutilized equipment, and increasingly complex supply chains. We believe the next generation of manufacturing will be enabled by more intelligent, flexible automation—systems that help people and machines work together to expand capacity and improve productivity.

We're building the intelligence layer that makes that possible.

Our team comes from the intersection of defense technology and AI. We've built and shipped products to DoD customers, and we bring deep experience across ML, robotics, product development, and defense.

We're a small, in-person team in New York City working on hard problems that matter. If you want to build the future of American manufacturing, we'd love to hear from you.

ABOUT THE ROLE

We are building intelligent robots that can perceive the world, understand instructions, and perform useful physical tasks in dynamic, real-world environments.

We are looking for a Staff Machine Learning Engineer to develop and scale the vision-language-action models that power our robots. You will own major parts of the model development lifecycle—from dataset construction and training recipes to distributed training infrastructure, evaluation, and on-robot validation.

This is a hands-on role at the intersection of machine learning research, large-scale systems, and robotics. You will work closely with researchers, robotics engineers, data teams, and operators to turn new modeling ideas into reliable robot capabilities. The right person is equally comfortable investigating why a policy failed on a robot, designing the next training experiment, and improving the infrastructure required to run that experiment efficiently at scale.

WHAT YOU’LL DO

• Train and improve vision-language-action models for robotic manipulation and other embodied tasks.
• Develop model architectures and training recipes spanning imitation learning, behavior cloning, transformer- and diffusion-based policies, multimodal pretraining, and post-training.
• Build scalable pipelines for pretraining, fine-tuning, evaluation, checkpointing, and model release.
• Train models across multi-node GPU clusters while improving utilization, throughput, stability, and cost efficiency.
• Design data mixtures, sampling strategies, augmentations, and curriculum approaches for large, heterogeneous robot datasets.
• Develop data loaders and preprocessing systems for synchronized video, language, robot state, actions, and other sensor modalities.
• Create reproducible experimentation systems, including configuration management, dataset and model versioning, experiment tracking, and automated regression testing.
• Define offline and on-robot metrics that measure task success, generalization, robustness, latency, safety, and failure modes.
• Build tools for inspecti

[…]
tack, from input data and training dynamics to inference and robot behavior.

• Experience independently owning technically ambiguous projects and delivering working systems.
• Clear communication skills and an ability to collaborate across research, infrastructure, robotics, hardware, and operations teams.
• A bachelor’s degree in computer science, machine learning, robotics, electrical engineering, or a related field—or equivalent practical experience.

NICE TO HAVE

• Experience developing vision-language-action models, vision-language models, or robotics foundation models.
• Experience with imitation learning, reinforcement learning, behavior cloning, diffusion policies, action tokenization, or action-conditioned world models.
• Experience training policies using data from multiple robot embodiments, task domains, or sensor configurations.
• Familiarity with large-scale pretraining, post-training, parameter-efficient fine-tuning, distillation, or model adaptation.
• Experience optimizing distributed training using techniques such as FSDP, tensor or pipeline parallelism, mixed precision, gradient checkpointing, or sharded data loading.
• Familiarity with robotics middleware and simulation environments such as ROS/ROS 2, MuJoCo, Isaac Sim, or Gazebo.
• Experience deploying and evaluating learned policies on physical robotic systems.
• Publications or meaningful open-source contributions in machine learning, robotics, computer vision, or related areas.

Nidus Technologies is an equal opportunity employer, and qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender perception or identity, national origin, age, marital status, protected veteran status, or disability status.

We encourage candidates from all backgrounds to apply, even if you don't feel like you're a perfect fit. If you're passionate about contributing to our mission, we'd love to hear from you!