PyPI page
Home page
Author:
None
Summary:
Typed tensor-shaping, masking, padding, device-routing, and checkpoint utilities for PyTorch and `einops`. A superset of `lucidrains/torch-einops-utils` with similar utilities from other lucidrains packages. Adds strict typing, extensive tests, and comprehensive docstrings.
Latest version:
0.1.9
Required dependencies:
einops
|
huntermakespy
|
torch
|
typing-extensions
Optional dependencies:
more-itertools
|
numpy
|
pytest
|
pytest-cov
Downloads last day:
10
Downloads last week:
76
Downloads last month:
522