psoulos / dtmLinks
Differentiable Tree Machine
☆13Updated last year
Alternatives and similar repositories for dtm
Users that are interested in dtm are comparing it to the libraries listed below
Sorting:
- The Energy Transformer block, in JAX☆57Updated last year
- Sequence Modeling with Structured State Spaces☆64Updated 2 years ago
- 🧮 Algebraic Positional Encodings.☆13Updated 5 months ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- ☆34Updated 2 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- A PyTorch implementation of Bayesian flow networks (Graves et al., 2023).☆26Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆62Updated 2 months ago
- Implementation of approximate free-energy minimization in PyTorch☆19Updated 3 years ago
- ☆17Updated 9 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆55Updated 3 months ago
- Experiments on the impact of depth in transformers and SSMs.☆30Updated 7 months ago
- Code for the paper https://arxiv.org/abs/2205.14987v2☆51Updated last year
- Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…☆10Updated 7 months ago
- Stochastic Normalizing Flows☆76Updated 3 years ago
- ☆32Updated last year
- Official implementation of the NeurIPS 24 paper of statistical flow matching (SFM) for discrete generation.☆26Updated 7 months ago
- ☆53Updated 8 months ago
- Implementation of deep implicit attention in PyTorch☆65Updated 3 years ago
- JAX exponential map normalising flows on sphere☆17Updated 4 years ago
- Graphically structured diffusion model.☆20Updated last year
- Code for "Discovering Non-monotonic Autoregressive Orderings with Variational Inference" (paper and code updated from ICLR 2021)☆12Updated last year
- Squared Non-monotonic Probabilistic Circuits☆22Updated 4 months ago
- Scalable and Stable Parallelization of Nonlinear RNNS☆15Updated 4 months ago
- Scalable training and inference for Probabilistic Circuits☆65Updated 2 weeks ago
- Code for the paper https://arxiv.org/abs/2402.04997☆78Updated last year
- ☆27Updated last year
- This is the code repository associated with the paper "Abstractors and relational cross-attention: An inductive bias for explicit relati…☆17Updated 9 months ago
- ☆34Updated 2 months ago
- Code for Pushdown Layers from our EMNLP 2023 paper☆28Updated last year