psoulos / dtmLinks
Differentiable Tree Machine
☆13Updated 2 years ago
Alternatives and similar repositories for dtm
Users that are interested in dtm are comparing it to the libraries listed below
Sorting:
- Transformers with doubly stochastic attention☆53Updated 3 years ago
- Sequence Modeling with Structured State Spaces☆67Updated 3 years ago
- 🧮 Algebraic Positional Encodings.☆18Updated 5 months ago
- ☆18Updated last year
- Scalable and Stable Parallelization of Nonlinear RNNS☆28Updated 3 months ago
- ☆42Updated 3 years ago
- Graphically structured diffusion model.☆21Updated 2 years ago
- The Energy Transformer block, in JAX☆63Updated 2 years ago
- ☆32Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆71Updated 10 months ago
- This is the official code release for Bayesian Flow Networks.☆309Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Updated 11 months ago
- Implementations of various linear RNN layers using pytorch and triton☆54Updated 2 years ago
- Code for the paper https://arxiv.org/abs/2205.14987v2☆58Updated last year
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆70Updated 11 months ago
- Squared Non-monotonic Probabilistic Circuits☆25Updated last year
- ☆62Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆190Updated 3 weeks ago
- Accelerated First Order Parallel Associative Scan☆196Updated last month
- Implementation of Action Matching☆51Updated 2 years ago
- ☆33Updated last year
- Perceiver (transformer variant) implemented in JAX and Flax☆13Updated 4 years ago
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆55Updated 5 months ago
- Scalable training and inference for Probabilistic Circuits☆92Updated this week
- Simple Scalable Discrete Diffusion for text in PyTorch☆37Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆109Updated 3 months ago
- A State-Space Model with Rational Transfer Function Representation.☆83Updated last year
- ☆40Updated 2 years ago
- A PyTorch implementation of Bayesian flow networks (Graves et al., 2023).☆29Updated 2 years ago