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.

GitHub

1k stars
16 watching
34 forks
Language: Python
last commit: 4 months ago
Linked from 1 awesome list

named-tensorspython-typingpytorchshapetensorstyping

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
wkentaro/pytorch-for-numpy-users Provides a PyTorch interface for working with NumPy arrays and tensors. 687
stewori/pytypes Provides runtime typechecking and annotation tools for Python to enforce strict type compatibility 200
stared/thinking-in-tensors-writing-in-pytorch A hands-on introduction to deep learning using PyTorch, explaining mathematical concepts through code examples 329
tensorly/tensorly A Python library providing tools and methods for working with tensors, including decomposition and learning techniques. 1,566
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,192
felixgwu/img_classification_pk_pytorch A PyTorch project for comparing image classification models and facilitating quick experiment setup 365
pytorch-labs/torchfix A tool to analyze and fix issues in PyTorch-related Python code, with automated fixes available. 102
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 413
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 83,959
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 103
davidfritzsche/pytest-mypy-testing A tool to verify mypy's static type analysis output 30
ethanhs/python-typecheckers A comprehensive collection of Python type checkers, categorized by syntax and execution time. 62