locuslab / torchdeq
Modern Fixed Point Systems using Pytorch
☆90Updated last year
Alternatives and similar repositories for torchdeq
Users that are interested in torchdeq are comparing it to the libraries listed below
Sorting:
- NF-Layers for constructing neural functionals.☆84Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆83Updated 2 months ago
- ☆156Updated 9 months ago
- ☆142Updated last year
- Code release for the paper Generative Modeling With Inverse Heat Dissipation☆69Updated 2 years ago
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆104Updated 5 months ago
- ☆60Updated 2 years ago
- Flow-matching algorithms in JAX☆90Updated 9 months ago
- ☆149Updated 2 years ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆154Updated 2 months ago
- Implementation of "Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains" by Tancik et al.☆91Updated 2 years ago
- NeurIPS23 "Flow Factorized Representation Learning"☆35Updated 6 months ago
- Collecting research materials on neural samplers with diffusion/flow models☆46Updated 2 months ago
- Code for the paper: Rotating Features for Object Discovery☆51Updated 9 months ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆29Updated last year
- Implicit networks can be trained efficiently and simply by using Jacobian-free Backprop (JFB).☆35Updated 3 years ago
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆146Updated 3 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Score-based generative models for compact manifolds☆110Updated last year
- ☆70Updated 2 years ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Code for reproducing results in the sliced score matching paper (UAI 2019)☆148Updated 5 years ago
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆77Updated last week
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆79Updated 9 months ago
- Use Jax functions in Pytorch☆240Updated last year
- ☆180Updated 5 months ago
- code for "Riemannian Flow Matching on General Geometries".☆224Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆123Updated last year
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- Official implementation of Transformer Neural Processes☆76Updated 2 years ago