On Jetson Thor, we see speedups about 1.2x to 7.9x from runtime optimizations alone and up to 33.78x for LingBot-VA when we combine those runtime optimizations with a distilled few-step diffusion scheduler, going from the original 25 visual / 50 action steps to 2 / 4 steps. Across 50 Robotwin2.0 tasks, we evaluated 1,153 episodes per configuration, LingBot-VA with InstinctFlash at 2 visual / 4 action steps achieved a 90.5% success rate, compared with 92.1% for the baseline at 25 visual / 50 action steps.
Here’s an optimized 5B world action model, running in real time on a Jetson Thor: https://youtu.be/nku65iyL5Fw
InstinctFlash currently supports 8 VLA / world-action model families, including pi0.5 and NVIDIA Cosmos Policy, across RTX 4090 / 5090 and Jetson Thor.
Just give it your fine-tuned checkpoint and InstinctFlash handles the rest, exposing the accelerated model through a Python runtime or an OpenPI-compatible WebSocket server.
We started working on this because we kept running into the same problem while deploying robot policies, the models were getting much better, but inference was often way too slow for the control loop we actually wanted.
For pi0.5, mixed-precision GEMMs and CUDA graphs speed up computation and reduce launch overhead. For Cosmos, caching avoids redundant computation across diffusion steps. World-action models’ diffusion denoising step depends on the previous one which motivated our work on few-step distillation.
Right now, InstinctFlash contains 6 aspects of optimization.
- Graph: CUDA graph capture, memory planning and separating prefill from repeated execution.
- Cache: Reusing KV and conditioning state across diffusion steps and prediction calls.
- Attention: Specialized attention paths for different model architectures.
- Kernels: Fused operations and kernels tailored to specific backends and tensor layouts.
- Precision: FP8 and mixed-precision execution.
- Model: Few-step distillation for diffusion and action generation.
Teams at Samsung, Siemens, and other robotics startups have used InstinctFlash for model acceleration on VLAs, WAMs, and diffusion-based world models. Now we are opening up access to you.
Feel free to try it here: https://github.com/General-Instinct/InstinctFlash
More implementation details and benchmarks: https://general-instinct.com/blog/instinctflash-edge-inferen...
Would love to hear your feedback!