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:
- ☆33Jan 14, 2021Updated 5 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆103Jul 6, 2023Updated 2 years ago
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- ☆153May 25, 2020Updated 5 years ago
- Exponential families for JAX☆76Feb 20, 2026Updated last week
- Live Python Notebooks with any Editor☆279Dec 7, 2022Updated 3 years ago
- Silly twitter torch implementations.☆46Oct 14, 2022Updated 3 years ago
- Bayesian inference for a logistic regression model in various languages☆43Jul 12, 2023Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆47Oct 26, 2020Updated 5 years ago
- A framework for implementing equivariant DL☆10May 25, 2021Updated 4 years ago
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 2 years ago
- ☆30Feb 11, 2022Updated 4 years ago
- Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"☆13Oct 25, 2021Updated 4 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- Shape annotations for numpy arrays and pytorch/tensorflow tensors.☆14Jul 1, 2022Updated 3 years ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆30Jan 22, 2024Updated 2 years ago
- ☆774Jan 27, 2024Updated 2 years ago
- ☆31Mar 10, 2020Updated 5 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- Normalizing Flows using JAX☆86Nov 30, 2023Updated 2 years ago
- a high performance system for customized-precision distributed deep learning☆12Dec 10, 2020Updated 5 years ago
- Sharing Deep Learning Research Models with Lightning Part 1: Building A Super Resolution App☆16Jun 30, 2022Updated 3 years ago
- "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.☆14Jan 23, 2024Updated 2 years ago
- ☆15Mar 19, 2024Updated last year
- This project was moved to: https://github.com/coax-dev/coax☆161Nov 28, 2022Updated 3 years ago
- Implicit generative models and related stuff based on the MMD, in PyTorch☆16Sep 24, 2020Updated 5 years ago
- An empirical investigation of deep learning theory☆16Oct 3, 2019Updated 6 years ago
- Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)☆268May 18, 2020Updated 5 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆111Jul 19, 2024Updated last year
- Concise deep learning for JAX☆183Oct 21, 2020Updated 5 years ago
- Tree-structured recurrent switching linear dynamical systems☆38Jul 13, 2020Updated 5 years ago
- ☆620Jan 23, 2026Updated last month
- Scalable distributed reinforcement learning agents on kubernetes☆57Jul 5, 2023Updated 2 years ago
- ☆15Dec 28, 2020Updated 5 years ago
- ☆18Mar 20, 2022Updated 3 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Jan 16, 2023Updated 3 years ago
- CLU lets you write beautiful training loops in JAX.☆366Feb 20, 2026Updated last week
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated 11 months ago