torchtyping
Tensor checker
Provides type annotations and dynamic checking for tensor shapes and types in PyTorch
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
1k stars
16 watching
34 forks
Language: Python
last commit: 9 months ago
Linked from 1 awesome list
named-tensorspython-typingpytorchshapetensorstyping
Related projects:
| Repository | Description | Stars |
|---|---|---|
 wkentaro/pytorch-for-numpy-users | Provides a PyTorch interface for working with NumPy arrays and tensors. | 691 |
 stewori/pytypes | Provides runtime typechecking and annotation tools for Python to enforce strict type compatibility | 201 |
 stared/thinking-in-tensors-writing-in-pytorch | A hands-on introduction to deep learning using PyTorch, explaining mathematical concepts through code examples | 331 |
 tensorly/tensorly | A Python library providing tools and methods for working with tensors, including decomposition and learning techniques. | 1,576 |
 mariusvniekerk/pytest-monkeytype | A pytest plugin that generates annotations and stubs for Python modules based on their type definitions | 43 |
 detachhead/basedpyright | An improved version of Pyright with enhanced type checking and support for popular IDEs | 1,300 |
 felixgwu/img_classification_pk_pytorch | A PyTorch project for comparing image classification models and facilitating quick experiment setup | 366 |
 pytorch-labs/torchfix | A tool to analyze and fix issues in PyTorch-related Python code, with automated fixes available. | 111 |
 jalola/compare-tensorflow-pytorch | A testing module for comparing the implementation of specific layers between TensorFlow and Pytorch | 72 |
 laurentmazare/ocaml-torch | Bindings for PyTorch's tensor library in OCaml for GPU acceleration and automatic differentiation | 416 |
 pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 84,978 |
 nearai/torchfold | A PyTorch module for dynamic batching and optimized computation on deep neural networks | 221 |
 typeddjango/pytest-mypy-plugins | A plugin for testing type checking and stubs using mypy and pytest | 107 |
 davidfritzsche/pytest-mypy-testing | A tool to verify mypy's static type analysis output | 31 |