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:
- Easy API deployment for Heroku☆14Mar 6, 2023Updated 3 years ago
- ☆33Jan 14, 2021Updated 5 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆121Jun 5, 2022Updated 4 years ago
- A functional training loops library for JAX☆87Feb 13, 2024Updated 2 years ago
- ☆80Sep 4, 2021Updated 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.
- A Pytree Module system for Deep Learning in JAX☆210Feb 26, 2023Updated 3 years ago
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆14Jul 2, 2023Updated 3 years ago
- ☆154May 25, 2020Updated 6 years ago
- Composable kernels for scikit-learn implemented in JAX.☆47Oct 26, 2020Updated 5 years ago
- Exponential families for JAX☆77Updated this week
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆22Jul 20, 2024Updated 2 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☆103Jul 6, 2023Updated 3 years ago
- Silly twitter torch implementations.☆48Oct 14, 2022Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- A JAX implementation of stochastic addition.☆13Aug 15, 2022Updated 4 years ago
- A framework for implementing equivariant DL☆10May 25, 2021Updated 5 years ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆31Jan 22, 2024Updated 2 years ago
- Interactive textbook on state space models☆12Apr 11, 2022Updated 4 years ago
- ☆774Jan 27, 2024Updated 2 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 6 years ago
- Content, code, and resources for the book How to Train Your Robot.☆25Dec 2, 2023Updated 2 years ago
- ☆15Mar 19, 2024Updated 2 years ago
- ☆30Feb 11, 2022Updated 4 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Concise deep learning for JAX☆182Oct 21, 2020Updated 5 years ago
- fun visualization scripts☆10Updated this week
- This project was moved to: https://github.com/coax-dev/coax☆162Nov 28, 2022Updated 3 years ago
- CLU lets you write beautiful training loops in JAX.☆371Jul 7, 2026Updated 2 months ago
- Tree-structured recurrent switching linear dynamical systems☆39Jul 13, 2020Updated 6 years ago
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 3 years ago
- Documentation:☆132May 22, 2023Updated 3 years ago
- PIX is an image processing library in JAX, for JAX.☆448Sep 9, 2026Updated last week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Why multiple dispatch lets you write composable code☆40Sep 25, 2020Updated 5 years ago
- PyTorch implementations of Self-Supervised Learning Method☆13Feb 9, 2023Updated 3 years ago
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆290Nov 25, 2021Updated 4 years ago
- ☆655Updated this week
- ☆15Dec 28, 2020Updated 5 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Nov 9, 2019Updated 6 years ago
- Pytorch Lightning seed project with hydra☆18Oct 8, 2020Updated 5 years ago