EZTorch

Utility functions and fluent interface for PyTorch tensors and modules

Installation

PIP

# Latest release
pip install ez-torch

# Dev version 
pip install git+https://github.com/ichko/ez-torch

From repo

git clone https://github.com/ichko/ez-torch
cd ez-torch
poetry install
poetry run pytest

...