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:
- The Energy Transformer block, in JAX☆58Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- nanoGPT using Equinox☆13Updated 2 years ago
- flexible meta-learning in jax☆14Updated last year
- If it quacks like a tensor...☆58Updated 7 months ago
- ☆17Updated 10 months ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Gradient-based constrained optimization for JAX☆32Updated 2 years ago
- JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"☆19Updated 2 years ago
- ☆60Updated 3 years ago
- Code for minimum-entropy coupling.☆32Updated 11 months ago
- ☆39Updated 3 years ago
- Turn jitted jax functions back into python source code☆22Updated 6 months ago
- Flexible Inference for Predictive Coding Networks in JAX.☆47Updated 2 weeks ago
- General Modules for JAX☆65Updated 2 months ago
- ☆26Updated 2 years ago
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆22Updated 2 weeks ago
- JAX Arrays for human consumption☆93Updated this week
- ☆31Updated 7 months ago
- LoRA for arbitrary JAX models and functions☆138Updated last year
- Generative cellular automaton-like learning environments for RL.☆19Updated 4 months ago
- Code for the paper "Function-Space Learning Rates"☆20Updated 3 weeks ago
- Pytorch-like dataloaders for JAX.☆87Updated 3 weeks ago
- ☆30Updated 9 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- ☆43Updated 3 weeks ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆20Updated last year
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆32Updated 3 years ago
- Neural Networks for JAX☆84Updated 9 months ago
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function☆13Updated 2 years ago