smonsays / jax-hypernetworkLinks
A simple hypernetwork implementation in jax using haiku.
☆23Updated 2 years ago
Alternatives and similar repositories for jax-hypernetwork
Users that are interested in jax-hypernetwork are comparing it to the libraries listed below
Sorting:
- flexible meta-learning in jax☆14Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Gradient-based constrained optimization for JAX☆32Updated 2 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆53Updated last month
- ☆17Updated 9 months ago
- ☆39Updated 3 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- nanoGPT using Equinox☆13Updated 2 years ago
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆22Updated 2 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- Generative cellular automaton-like learning environments for RL.☆19Updated 4 months ago
- Code for the paper "Function-Space Learning Rates"☆20Updated last month
- ☆43Updated this week
- If it quacks like a tensor...☆58Updated 6 months ago
- ☆34Updated 3 years ago
- ☆32Updated 8 months ago
- ☆26Updated 2 years ago
- Pytrees + dataclasses ❤️☆63Updated last month
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Image augmentation library for Jax☆39Updated last year
- Turn jitted jax functions back into python source code☆22Updated 5 months ago
- ☆60Updated 3 years ago
- JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"☆19Updated last year
- Einsum-like high-level array sharding API for JAX☆34Updated 10 months ago
- ☆40Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated last week
- Flexible Inference for Predictive Coding Networks in JAX.☆45Updated this week
- The Energy Transformer block, in JAX☆56Updated last year
- JAX Arrays for human consumption☆91Updated last year