SDML-KU / qkvflowLinks
Neural ODE Transformers (ICLR 2025)
☆15Updated last month
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☆58Updated 4 months ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆81Updated 6 months ago
- Neural Diffusion Processes☆81Updated last year
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆66Updated 2 months ago
- Variational Autoencoder with non-euclidean (hyperbolic) latent space☆11Updated 2 years ago
- Code release for "Stochastic Optimal Control Matching"☆37Updated 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
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Code repository for Trajectory Flow Matching☆85Updated 11 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆103Updated 2 months ago
- Official PyTorch implementation of NeuralSVD (ICML 2024)☆20Updated last year
- FID computation in Jax/Flax.☆28Updated last year
- Fractional White Noises for Neural Stochastic Differential Equations (NeurIPS 2022)☆16Updated 2 years ago
- ☆33Updated 10 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆51Updated last year
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆110Updated last month
- Flow-matching algorithms in JAX☆105Updated last year
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆23Updated last year
- ☆27Updated 2 years ago
- Official implementation of Transformer Neural Processes☆78Updated 3 years ago
- Hierarchical State Space Models☆47Updated last year
- Official code release of Hilbert Diffusion Model (PyTorch ver.)☆18Updated last year
- PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.☆86Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆88Updated last year
- Official implementation of "Fourier Head: Helping Large Language Models Learn Complex Probability Distributions" (ICLR 2025)☆66Updated 6 months ago
- ☆120Updated 4 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- C++ and Cuda ops for fused FourierKAN☆80Updated last year
- Experiments for Neural Flows paper☆98Updated 3 years ago