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 3 years ago
- Pytrees + dataclasses ❤️☆63Updated 2 months ago
- ☆21Updated last year
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- AdaCat☆49Updated 2 years ago
- ☆60Updated 3 years ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- ☆15Updated 4 years ago
- Document parameters using comments☆10Updated 3 years ago
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12Updated 4 years ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 8 months ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆45Updated last year
- A snappy + easy + pretty TUI debugger for Python.☆46Updated 3 weeks ago
- Understanding RL vision Distill article☆23Updated 2 years ago
- Jax SSM Library☆49Updated 2 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 8 months ago
- Generative cellular automaton-like learning environments for RL.☆19Updated 4 months ago
- NumPy+Jax with named axes and an uncompromising attitude☆20Updated 3 months ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆54Updated last month
- ☆104Updated last year
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- RL agent to play μRTS with Stable-Baselines3 and PyTorch☆26Updated 3 years ago
- Performant, differentiable reinforcement learning☆121Updated last week