SDML-KU / qkvflowLinks
Neural ODE Transformers (ICLR 2025)
☆17Updated 4 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☆61Updated 8 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- Course Website for "AI618: Generative Model and Unsupervised Learning"☆37Updated 2 years ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆72Updated 2 months ago
- Neural Diffusion Processes☆82Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆107Updated 3 months ago
- ☆23Updated 11 months ago
- ☆28Updated 2 years ago
- Flow-matching algorithms in JAX☆114Updated last year
- A simple implementation of classifier-free guidance DDIM on MNIST☆36Updated 3 years ago
- Experiment with diffusion models that you can run on your local jupyter instances☆65Updated last year
- Implementation of Diffusion Transformer Model in Pytorch☆73Updated 3 weeks ago
- Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).☆70Updated 3 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆126Updated 5 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆51Updated last year
- Code release for "Stochastic Optimal Control Matching"☆39Updated last year
- ☆35Updated last year
- ☆123Updated 7 months ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆22Updated 2 years ago
- Hierarchical State Space Models☆49Updated last year
- Official code release of Hilbert Diffusion Model (PyTorch ver.)☆20Updated last year
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Updated last year
- A variational auto-encoder (VAE) framework with a new type of prior "Variational Mixture of Posteriors" prior, or VampPrior for short.☆10Updated 4 years ago
- ☆35Updated last year
- Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise☆40Updated last year
- My take on Flow Matching☆90Updated last year
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆82Updated last year
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆103Updated 4 months ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆82Updated 9 months ago
- Official implementation of Transformer Neural Processes☆78Updated 3 years ago