SDML-KU / qkvflowLinks
Neural ODE Transformers (ICLR 2025)
☆12Updated 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 Diffusion Processes☆81Updated 10 months ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆77Updated 2 months ago
- Code release for "Stochastic Optimal Control Matching"☆35Updated 10 months ago
- Official implementation of Transformer Neural Processes☆78Updated 2 years ago
- Fractional White Noises for Neural Stochastic Differential Equations (NeurIPS 2022)☆16Updated 2 years ago
- [ICLR 2022] Path integral sampler☆47Updated last year
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 9 months ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆62Updated 3 months ago
- Neural Optimal Transport with Lagrangian Costs☆56Updated last month
- ☆13Updated 2 years ago
- ☆32Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- ☆34Updated 2 months ago
- ☆31Updated 7 months ago
- ☆21Updated last year
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆30Updated 3 months ago
- ☆24Updated last year
- ☆11Updated 2 years ago
- Denoising Hamiltonian Network for Physical Reasoning☆32Updated 3 months ago
- [ICML2022] Variational Wasserstein gradient flow☆23Updated 2 years ago
- ☆28Updated last year
- ☆44Updated last year
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Code repository for Trajectory Flow Matching☆69Updated 7 months ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆29Updated last year
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆19Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated 5 months ago
- Flow-matching algorithms in JAX☆97Updated 10 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…☆29Updated last month
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆16Updated 11 months ago