JIT
// PyTorch implementation of JiT https://arxiv.org/abs/2511.13720
JiT
PyTorch implementation of JiT https://arxiv.org/abs/2511.13720
13EmergingUnknown
What it does
This is a PyTorch/GPU re-implementation of the paper Back to Basics: Let Denoising Generative Models Denoise: JiT adopts a minimalist and self-contained design for pixel-level high-resolution image diffusion. The original implementation was in JAX+TPU. This re-implementation is in PyTorch+GPU. Download ImageNet dataset, and place it in your . Download the code: A suitable conda environment
Getting Started
git
git clone https://github.com/LTH14/JiT