akandykeller / FERNNLinks
Official repository for the paper "Flow Equivariant Recurrent Neural Networks"
☆20Updated 4 months ago
Alternatives and similar repositories for FERNN
Users that are interested in FERNN are comparing it to the libraries listed below
Sorting:
- Flow-matching algorithms in JAX☆106Updated last year
- ☆120Updated 4 months ago
- ☆44Updated 2 months ago
- ☆34Updated 11 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆105Updated last week
- Neural Optimal Transport with Lagrangian Costs☆60Updated 5 months ago
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆111Updated 2 weeks ago
- ☆73Updated last year
- Flash Attention Triton kernel with support for second-order derivatives☆106Updated 2 weeks ago
- ☆53Updated last year
- My take on Flow Matching☆82Updated 9 months ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- NF-Layers for constructing neural functionals.☆90Updated last year
- Use Jax functions in Pytorch☆253Updated 2 years ago
- ☆44Updated last year
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆38Updated last year
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- Modern Fixed Point Systems using Pytorch☆121Updated 2 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆88Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.☆30Updated last year
- A simple library for scaling up JAX programs☆144Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 9 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆67Updated 2 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated 2 years ago
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆86Updated 3 weeks ago
- Tools and Utils for Experiments (TUX)☆15Updated 9 months ago
- Explorations into the recently proposed Taylor Series Linear Attention☆99Updated last year
- ☆34Updated 2 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆138Updated 8 months ago