SDML-KU / qkvflowLinks
Neural ODE Transformers (ICLR 2025)
☆16Updated 2 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 5 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- ☆34Updated 11 months ago
- 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☆67Updated 3 months ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆81Updated 7 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆105Updated 3 weeks ago
- ☆34Updated last year
- Neural Diffusion Processes☆81Updated last year
- Hierarchical State Space Models☆47Updated last year
- FID computation in Jax/Flax.☆29Updated last year
- ☆119Updated 5 months ago
- Flow-matching algorithms in JAX☆107Updated last year
- Implementation of Diffusion Transformer Model in Pytorch☆71Updated 6 months ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆20Updated 2 years ago
- Code repository for Trajectory Flow Matching☆89Updated last year
- A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with sim…☆36Updated 6 months ago
- Official implementation of Transformer Neural Processes☆78Updated 3 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆89Updated last year
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.☆85Updated last year
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆20Updated last year
- My take on Flow Matching☆84Updated 10 months ago
- ☆27Updated 2 years ago
- Variational Autoencoder with non-euclidean (hyperbolic) latent space☆11Updated 2 years ago
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆23Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- ☆34Updated 7 months ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆103Updated last month
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆28Updated last year