Framework-agnostic library for checking array/tensor shapes at runtime.
☆46May 4, 2021Updated 5 years ago
Alternatives and similar repositories for shapecheck
Users that are interested in shapecheck are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆33Jan 14, 2021Updated 5 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆120Jun 5, 2022Updated 4 years ago
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- An easy-to-use app to visualise attentions of various VQA models.☆41Oct 14, 2022Updated 3 years ago
- ☆80Sep 4, 2021Updated 4 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A Pytree Module system for Deep Learning in JAX☆212Feb 26, 2023Updated 3 years ago
- ☆154May 25, 2020Updated 6 years ago
- ☆31Mar 10, 2020Updated 6 years ago
- Composable kernels for scikit-learn implemented in JAX.☆47Oct 26, 2020Updated 5 years ago
- Live Python Notebooks with any Editor☆280Dec 7, 2022Updated 3 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆102Jul 6, 2023Updated 2 years ago
- Silly twitter torch implementations.☆48Oct 14, 2022Updated 3 years ago
- A framework for implementing equivariant DL☆10May 25, 2021Updated 5 years ago
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Interactive textbook on state space models☆12Apr 11, 2022Updated 4 years ago
- ☆773Jan 27, 2024Updated 2 years ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆31Jan 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 6 years ago
- Content, code, and resources for the book How to Train Your Robot.☆25Dec 2, 2023Updated 2 years ago
- ☆30Feb 11, 2022Updated 4 years ago
- ☆15Mar 19, 2024Updated 2 years ago
- Concise deep learning for JAX☆183Oct 21, 2020Updated 5 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆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☆39Jul 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 3 months ago
- Documentation:☆131May 22, 2023Updated 3 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.☆440Jun 2, 2026Updated 3 weeks ago
- Why multiple dispatch lets you write composable code☆40Sep 25, 2020Updated 5 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)☆268May 18, 2020Updated 6 years ago
- ☆640Jun 12, 2026Updated 2 weeks ago
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆289Nov 25, 2021Updated 4 years ago
- ☆15Dec 28, 2020Updated 5 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Nov 9, 2019Updated 6 years ago
- Shape annotations for numpy arrays and pytorch/tensorflow tensors.☆14Jul 1, 2022Updated 3 years ago
- An empirical investigation of deep learning theory☆16Oct 3, 2019Updated 6 years ago