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. 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).☆120Jun 5, 2022Updated 3 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
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆80Sep 4, 2021Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆212Feb 26, 2023Updated 3 years ago
- ☆153May 25, 2020Updated 5 years ago
- ☆31Mar 10, 2020Updated 6 years ago
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆14Jul 2, 2023Updated 2 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…☆20Jul 20, 2024Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆46Oct 26, 2020Updated 5 years ago
- Live Python Notebooks with any Editor☆279Dec 7, 2022Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- 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 4 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- ☆775Jan 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
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆30Feb 11, 2022Updated 4 years ago
- ☆15Mar 19, 2024Updated 2 years ago
- Concise deep learning for JAX☆183Oct 21, 2020Updated 5 years ago
- This project was moved to: https://github.com/coax-dev/coax☆161Nov 28, 2022Updated 3 years ago
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- ☆14Jun 12, 2023Updated 2 years ago
- CLU lets you write beautiful training loops in JAX.☆368Mar 3, 2026Updated last month
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 3 years ago
- Documentation:☆131May 22, 2023Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- PIX is an image processing library in JAX, for JAX.☆436Mar 6, 2025Updated last year
- Why multiple dispatch lets you write composable code☆40Sep 25, 2020Updated 5 years ago
- Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)☆268May 18, 2020Updated 5 years ago
- ☆634Apr 18, 2026Updated last week
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆288Nov 25, 2021Updated 4 years ago
- fun visualization scripts☆10Mar 1, 2026Updated last month
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Nov 9, 2019Updated 6 years ago