n2cholas / shapecheck
Framework-agnostic library for checking array/tensor shapes at runtime.
☆47Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for shapecheck
- ☆108Updated last year
- A JAX implementation of stochastic addition.☆12Updated 2 years ago
- ☆155Updated 4 years ago
- Silly twitter torch implementations.☆46Updated 2 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 3 years ago
- ☆33Updated 3 years ago
- ☆11Updated 4 years ago
- Functional machine learning for fun☆84Updated 3 years ago
- ☆30Updated 2 years ago
- Lightweight ML Experiment Logging 📖☆80Updated 2 months ago
- Image augmentation library for Jax☆37Updated 7 months ago
- Weekly humanitarian AI reading group at Mila.☆21Updated 5 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- 🧀 Pytorch code for the Fromage optimiser.☆122Updated 4 months ago
- Pytorch Lightning seed project with hydra☆18Updated 4 years ago
- Documentation:☆120Updated last year
- ☆58Updated 2 years ago
- ☆101Updated 3 years ago
- ☆67Updated last year
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- ☆97Updated 2 years ago
- ☆99Updated 4 months ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 3 years ago
- Codebase for Learning Invariances in Neural Networks☆94Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆37Updated 4 years ago
- ☆78Updated 3 years ago
- A Machine Learning workflow for Slurm.☆146Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆43Updated 3 years ago