Framework-agnostic library for checking array/tensor shapes at runtime.
☆46May 4, 2021Updated 4 years ago
Alternatives and similar repositories for shapecheck
Users that are interested in shapecheck are comparing it to the libraries listed below
Sorting:
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- ☆80Sep 4, 2021Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆212Feb 26, 2023Updated 3 years ago
- ☆31Mar 10, 2020Updated 6 years ago
- ☆153May 25, 2020Updated 5 years ago
- Exponential families for JAX☆76Mar 13, 2026Updated last week
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Jul 20, 2024Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆47Oct 26, 2020Updated 5 years ago
- Live Python Notebooks with any Editor☆279Dec 7, 2022Updated 3 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆103Jul 6, 2023Updated 2 years ago
- Silly twitter torch implementations.☆46Oct 14, 2022Updated 3 years ago
- A framework for implementing equivariant DL☆10May 25, 2021Updated 4 years ago
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- Interactive textbook on state space models☆12Apr 11, 2022Updated 3 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- ☆774Jan 27, 2024Updated 2 years ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆30Jan 22, 2024Updated 2 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Jan 16, 2023Updated 3 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- ☆30Feb 11, 2022Updated 4 years ago
- Concise deep learning for JAX☆183Oct 21, 2020Updated 5 years ago
- ☆18Mar 20, 2022Updated 4 years ago
- This project was moved to: https://github.com/coax-dev/coax☆161Nov 28, 2022Updated 3 years ago
- Tree-structured recurrent switching linear dynamical systems☆38Jul 13, 2020Updated 5 years ago
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- CLU lets you write beautiful training loops in JAX.☆367Mar 3, 2026Updated 2 weeks ago
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 2 years ago
- Documentation:☆129May 22, 2023Updated 2 years ago
- Bayesian inference for a logistic regression model in various languages☆43Jul 12, 2023Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated last year
- Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)☆268May 18, 2020Updated 5 years ago
- ☆627Jan 23, 2026Updated last month
- ☆15Dec 28, 2020Updated 5 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Nov 9, 2019Updated 6 years ago
- Jax-based MaxEnt☆17Nov 24, 2019Updated 6 years ago
- An empirical investigation of deep learning theory☆16Oct 3, 2019Updated 6 years ago
- Pytorch Lightning seed project with hydra☆18Oct 8, 2020Updated 5 years ago
- ☆24Jul 31, 2023Updated 2 years ago
- Implicit generative models and related stuff based on the MMD, in PyTorch☆16Sep 24, 2020Updated 5 years ago