PyPI page
Home page
Author:
Phil Wang, Hunter Hogan
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.2
Required dependencies:
einops
|
torch
|
typing-extensions
Downloads last day:
25
Downloads last week:
208
Downloads last month:
382