Metadata-Version: 2.4
Name: homa
Version: 0.1.3
Summary: A curated list of machine learning and deep learning helpers.
Author-email: Taha Shieenavaz <tahashieenavaz@gmail.com>
Requires-Python: >=3.7
Description-Content-Type: text/markdown
Requires-Dist: numpy
Requires-Dist: torch
Requires-Dist: fire

## Device Management

```py
from homa import cpu, mps, cuda, device

torch.tensor([1, 2, 3, 4, 5]).to(cpu())
torch.tensor([1, 2, 3, 4, 5]).to(cuda())
torch.tensor([1, 2, 3, 4, 5]).to(mps())
torch.tensor([1, 2, 3, 4, 5]).to(device())
```
