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☆60Updated 7 months ago
- Course Website for "AI618: Generative Model and Unsupervised Learning"☆37Updated 2 years ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- ☆28Updated 2 years ago
- Flow-matching algorithms in JAX☆114Updated last year
- ☆35Updated last year
- Neural Diffusion Processes☆81Updated last year
- ☆35Updated last year
- ☆23Updated 11 months ago
- Hierarchical State Space Models☆48Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆106Updated 2 months ago
- Code repository for Trajectory Flow Matching☆96Updated last year
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆70Updated last month
- Code release for "Stochastic Optimal Control Matching"☆39Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆71Updated 9 months ago
- A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with sim…☆40Updated 8 months ago
- Implementation of Diffusion Transformer Model in Pytorch☆73Updated this week
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆22Updated 2 years ago
- PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.☆87Updated last year
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆81Updated 8 months ago
- Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).☆67Updated 3 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆92Updated last year
- Official code release of Hilbert Diffusion Model (PyTorch ver.)☆19Updated last year
- My take on Flow Matching☆89Updated 11 months ago
- ☆122Updated 6 months ago
- FID computation in Jax/Flax.☆29Updated last year
- A simple implementation of classifier-free guidance DDIM on MNIST☆36Updated 3 years ago
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆148Updated 8 months ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Updated last year
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆20Updated last year