araffin / datasaurustLinks
Blazingly fast implementation of the Datasaurus paper. Same Stats, Different Graphs.
☆19Updated 2 years ago
Alternatives and similar repositories for datasaurust
Users that are interested in datasaurust are comparing it to the libraries listed below
Sorting:
- Clockwork VAEs in JAX/Flax☆32Updated 4 years ago
- RL agent to play μRTS with Stable-Baselines3 and PyTorch☆26Updated 3 years ago
- AdaCat☆49Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- ☆21Updated last year
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 9 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- ☆60Updated 3 years ago
- Image augmentation library for Jax☆39Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- Gym environment for playing Wordle with RL agents☆39Updated 3 years ago
- Pytrees + dataclasses ❤️☆63Updated 2 months ago
- ☆28Updated 2 years ago
- Understanding RL vision Distill article☆23Updated 2 years ago
- Performant, differentiable reinforcement learning☆121Updated this week
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- A framework for implementing equivariant DL☆10Updated 4 years ago
- ☆23Updated 3 years ago
- Generalised UDRL☆37Updated 3 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆45Updated last year
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- Analogous Safe-state Exploration (ASE) is an algorithm for provably safe and optimal exploration in MDPs with unknown, stochastic dynamic…☆11Updated 4 years ago
- Variational Reinforcement Learning☆16Updated 11 months ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago
- ☆20Updated 6 years ago
- Plannable Approximations to MDP Homomorphisms: Equivariance under Actions☆30Updated 5 years ago
- Jax SSM Library☆49Updated 2 years ago
- Opinionated library for managing hyperparameters and mutable state of machine learning training systems.☆19Updated last year