SDML-KU / qkvflowLinks
Neural ODE Transformers (ICLR 2025)
☆14Updated 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☆56Updated last month
- Neural Diffusion Processes☆81Updated 11 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 9 months ago
- Lightning-like training API for JAX with Flax☆42Updated 7 months ago
- Denoising Hamiltonian Network for Physical Reasoning☆32Updated 3 months ago
- Official implementation of Transformer Neural Processes☆78Updated 2 years ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆78Updated 3 months ago
- PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.☆82Updated last year
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆21Updated last year
- ☆35Updated 3 months ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆30Updated 4 months ago
- Programming for AI Practice☆14Updated 4 years ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆65Updated last month
- Fractional White Noises for Neural Stochastic Differential Equations (NeurIPS 2022)☆16Updated 2 years ago
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆27Updated last year
- Flow-matching algorithms in JAX☆97Updated 11 months ago
- Course Website for "AI618: Generative Model and Unsupervised Learning"☆38Updated 2 years ago
- ☆31Updated 7 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated 11 months ago
- Code release for "Stochastic Optimal Control Matching"☆36Updated 11 months ago
- Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"☆20Updated last year
- Official code release of Hilbert Diffusion Model (PyTorch ver.)☆17Updated 11 months ago
- ☆24Updated last year
- Pytorch-like dataloaders for JAX.☆90Updated last month
- A minimal implementation of Gaussian process regression in PyTorch☆63Updated 2 years ago
- Code repository for Trajectory Flow Matching☆73Updated 8 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆99Updated 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
- ☆13Updated 2 years ago
- ☆32Updated last year