The Energy Transformer block, in JAX
☆64Dec 14, 2023Updated 2 years ago
Alternatives and similar repositories for energy-transformer-jax
Users that are interested in energy-transformer-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Hierarchical Associative Memory User Experience☆108Jan 13, 2026Updated 2 months ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 8 months ago
- Official repository for the paper "Automating Continual Learning"☆18Jun 11, 2025Updated 9 months ago
- Repository for Sparse Universal Transformers☆20Oct 23, 2023Updated 2 years ago
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆16Nov 8, 2024Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- This is the official repository for the paper "Laplacian Features for Learning with Hyperbolic Space"☆13Aug 8, 2022Updated 3 years ago
- repo for code for paper on general theory associative memory models☆22Jun 15, 2022Updated 3 years ago
- ☆13Mar 21, 2023Updated 3 years ago
- Official code for Slot-Transformer for Videos (STEVE)