google-deepmind / fancyflags
A Python library for defining flat or nested dictionary flags.
☆30Updated last year
Related projects ⓘ
Alternatives and complementary repositories for fancyflags
- ☆99Updated 4 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- ☆78Updated 3 years ago
- Pytrees + dataclasses ❤️☆64Updated 4 months ago
- Tools for JAX☆43Updated this week
- ☆153Updated 11 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆37Updated 4 years ago
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)☆49Updated 2 months ago
- Image augmentation library for Jax☆37Updated 7 months ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- ☆72Updated last week
- JAX Arrays for human consumption☆88Updated last year
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Autoregressive Energy Machines☆77Updated last year
- Annotating tensor shapes using Python types☆158Updated last year
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Inference on non-linear dynamical systems written in JAX☆12Updated 4 years ago
- ☆24Updated 5 years ago
- JMP is a Mixed Precision library for JAX.☆187Updated 6 months ago
- A functional training loops library for JAX☆85Updated 9 months ago
- OpTree: Optimized PyTree Utilities☆152Updated this week
- Example python package with pybind11 cpp extension☆57Updated 3 years ago
- Exponential families for JAX☆55Updated this week
- A dead simple Python package for creating custom JAX pytree objects☆62Updated last week
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- A JAX implementation of stochastic addition.☆12Updated 2 years ago
- Normalizing Flows in Jax☆105Updated 4 years ago
- ☆71Updated this week