Keely-Ai/JAX-F2D2

JAX implementation of F2D2: Joint Distillation for Fast Likelihood Evaluation and Sampling in Flow-based Models (https://arxiv.org/abs/2512.02636)

语言：Jupyter Notebook · GitHub 星标：4 · Fork：0

在 GitHub 上查看原仓库