SDML-KU / qkvflowLinks
Neural ODE Transformers (ICLR 2025)
☆17Updated 3 months ago
Alternatives and similar repositories for qkvflow
Users that are interested in qkvflow are comparing it to the libraries listed below
Sorting:
- Neural Optimal Transport with Lagrangian Costs☆60Updated 6 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- Neural Diffusion Processes☆80Updated last year
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆68Updated 3 weeks ago
- ☆35Updated last year
- Variational Autoencoder with non-euclidean (hyperbolic) latent space☆12Updated 3 years ago
- ☆34Updated last year
- Flow-matching algorithms in JAX☆112Updated last year
- ☆27Updated 2 years ago
- Deep Networks Grok All the Time and Here is Why☆38Updated last year
- Code release for "Stochastic Optimal Control Matching"☆39Updated last year
- ☆35Updated 8 months ago
- Lightning-like training API for JAX with Flax☆44Updated last year
- Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).☆55Updated 2 months ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆22Updated 2 years ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆106Updated last month
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- Course Website for "AI618: Generative Model and Unsupervised Learning"☆37Updated 2 years ago
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆35Updated 8 months ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆84Updated last year
- A simple implementation of classifier-free guidance DDIM on MNIST☆35Updated 3 years ago
- Experiment with diffusion models that you can run on your local jupyter instances☆63Updated last year
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆81Updated 7 months ago
- ☆23Updated 10 months ago
- Implementation of Diffusion Transformer Model in Pytorch☆71Updated 7 months ago
- Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"☆43Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Hierarchical State Space Models☆47Updated last year
- Scalable Computation of Hessian Diagonals☆14Updated last year