parrt / tensor-sensor
The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
☆804Updated 2 years ago
Alternatives and similar repositories for tensor-sensor:
Users that are interested in tensor-sensor are comparing it to the libraries listed below
- Shape and dimension inference (Keras-like) for PyTorch layers and neural networks☆572Updated 2 years ago
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆695Updated last year
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,717Updated 2 weeks ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,420Updated 7 months ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,414Updated this week
- A PyTorch implementation of Neighbourhood Components Analysis.☆400Updated 4 years ago
- ML Collections is a library of Python Collections designed for ML use cases.☆928Updated last month
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆474Updated 2 years ago
- Pytorch Lightning code guideline for conferences☆1,258Updated last year
- Web-based dashboard for Sacred☆549Updated 2 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,078Updated this week
- Fast, differentiable sorting and ranking in PyTorch☆796Updated last year
- Deep Learning Experiment Management☆640Updated 2 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,092Updated last year
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 2 years ago
- A simplified framework and utilities for PyTorch☆571Updated 3 months ago
- Pytorch library for fast transformer implementations☆1,686Updated last year
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,177Updated this week
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,610Updated 2 years ago
- Python 3.8+ toolbox for submitting jobs to Slurm☆1,397Updated 6 months ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,228Updated this week
- Generic template to bootstrap your PyTorch project.☆642Updated last year
- Profiling and inspecting memory in pytorch☆1,049Updated 7 months ago
- YACS -- Yet Another Configuration System☆1,302Updated 2 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago
- Tips for releasing research code in Machine Learning (with official NeurIPS 2020 recommendations)☆2,660Updated last year
- High performance model preprocessing library on PyTorch☆650Updated 11 months ago
- torchview: visualize pytorch models☆892Updated last week
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,044Updated 2 weeks ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,112Updated 2 years ago