smonsays / jax-hypernetwork
A simple hypernetwork implementation in jax using haiku.
☆23Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for jax-hypernetwork
- If it quacks like a tensor...☆52Updated last week
- Flow-matching algorithms in JAX☆77Updated 3 months ago
- Pytorch-like dataloaders in JAX.☆59Updated last month
- ☆36Updated 2 years ago
- flexible meta-learning in jax☆12Updated last year
- The Energy Transformer block, in JAX☆53Updated 11 months ago
- Gradient-based constrained optimization for JAX☆26Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆38Updated last week
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆79Updated 9 months ago
- JAX Arrays for human consumption☆88Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- ☆26Updated 2 years ago
- Pytrees + dataclasses ❤️☆64Updated 4 months ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 2 years ago
- ☆58Updated 2 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆41Updated last month
- Generative cellular automaton-like learning environments for RL.☆19Updated last month
- JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"☆20Updated last year
- ☆40Updated 4 months ago
- ☆16Updated 2 months ago
- ☆48Updated 9 months ago
- ☆34Updated 3 years ago
- ☆46Updated last month
- A simple library for scaling up JAX programs☆127Updated 2 weeks ago
- ☆33Updated last year
- A functional training loops library for JAX☆85Updated 9 months ago
- ☆24Updated 5 years ago
- ☆29Updated 2 months ago
- Image augmentation library for Jax☆37Updated 7 months ago