jax-raft

github.comtool

JAX/Flax port of the RAFT optical flow estimator.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Lorax github.com

   tool category #jax link ↗

   Automatically apply LoRA to JAX models (Flax, Haiku, etc.)

   
2. ▲
   jax-flows github.com

   tool category #jax link ↗

   Normalizing flows in JAX.

   
3. ▲
   jax-resnet github.com

   tool category #jax link ↗

   Implementations and checkpoints for ResNet variants in Flax.

   
4. ▲
   purejaxrl github.com

   tool category #jax link ↗

   Vectorisable, end-to-end RL algorithms in JAX.

   
5. ▲
   FlaxVision github.com

   tool category #jax link ↗

   Flax version of TorchVision.

   
6. ▲
   DiffeRT github.com

   tool category #jax link ↗

   Differentiable Ray Tracing toolbox for Radio Propagation powered by the JAX ecosystem.

JAX › Libraries > New Libraries: “JAX/Flax port of the RAFT optical flow estimator.”