Archives AI News

How Lightricks trains video diffusion models at scale with JAX on TPU

Training large video diffusion models at scale isn’t just computationally expensive — it can become impossible when your framework can’t keep pace with your ambitions.  JAX has become a popular computational framework across AI applications, now recognized for its capabilities…