bhoov / energy-transformer-jaxLinks
The Energy Transformer block, in JAX
☆56Updated last year
Alternatives and similar repositories for energy-transformer-jax
Users that are interested in energy-transformer-jax are comparing it to the libraries listed below
Sorting:
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆22Updated 2 years ago
- ☆51Updated last year
- Scalable and Stable Parallelization of Nonlinear RNNS☆15Updated 4 months ago
- ☆32Updated 8 months ago
- We integrate discrete diffusion models with neurosymbolic predictors for scalable and calibrated learning and reasoning☆33Updated 2 weeks ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆26Updated 2 years ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- ☆17Updated 9 months ago
- ☆29Updated 2 months ago
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆24Updated last year
- ☆53Updated 8 months ago
- Implementation of PSGD optimizer in JAX☆33Updated 5 months ago
- ☆39Updated 3 years ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆82Updated 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 Scalable Approximate Method for Probabilistic Neurosymbolic Inference☆15Updated 4 months ago
- Hierarchical Associative Memory User Experience☆100Updated last year
- Bayesian model reduction for probabilistic machine learning☆11Updated last month
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆32Updated 3 years ago
- Flexible Inference for Predictive Coding Networks in JAX.☆45Updated this week
- Graphically structured diffusion model.☆20Updated last year
- Fast training of unitary deep network layers from low-rank updates☆28Updated 2 years ago
- ☆32Updated last year
- 🧮 Algebraic Positional Encodings.☆13Updated 4 months ago
- ☆9Updated 2 years ago
- ☆65Updated 11 months ago
- Code associated to papers on superposition (in ML interpretability)☆28Updated 2 years ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆84Updated 2 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆81Updated last year