homa 0.0.1__py3-none-any.whl → 0.0.11__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of homa might be problematic. Click here for more details.

homa/torch/Module.py ADDED
@@ -0,0 +1,8 @@
1
+ import torch
2
+ from ..device import get_device
3
+
4
+
5
+ class Module(torch.nn.Module):
6
+ def __init__(self):
7
+ super(Module, self).__init__()
8
+ self.to(get_device())
homa/torch/__init__.py ADDED
@@ -0,0 +1 @@
1
+ from .Module import Module
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: homa
3
- Version: 0.0.1
3
+ Version: 0.0.11
4
4
  Summary: A curated list of machine learning and deep learning helpers.
5
5
  Author-email: Taha Shieenavaz <tahashieenavaz@gmail.com>
6
6
  Requires-Python: >=3.7
@@ -8,5 +8,13 @@ Description-Content-Type: text/markdown
8
8
  Requires-Dist: numpy
9
9
  Requires-Dist: torch
10
10
 
11
- # homa
12
- A curated collection of machine learning and deep learning helper functions.
11
+ ## Device Management
12
+
13
+ ```py
14
+ from homa import cpu, mps, cuda, device
15
+
16
+ torch.tensor([1, 2, 3, 4, 5]).to(cpu())
17
+ torch.tensor([1, 2, 3, 4, 5]).to(cuda())
18
+ torch.tensor([1, 2, 3, 4, 5]).to(mps())
19
+ torch.tensor([1, 2, 3, 4, 5]).to(device())
20
+ ```
@@ -0,0 +1,9 @@
1
+ homa/__init__.py,sha256=NBYFKizG8UASiz5HLsEBqzXNGlWr78xm4sLr5hxKvjU,46
2
+ homa/device.py,sha256=9kKXfpYfnEk2cFQWPfcJrVloHgC_SSbP4I8IRY9TYk4,343
3
+ homa/settings.py,sha256=CPZDPvs1380O7SY7FcSKol8kBVFVVYFgSJl3YEyJuZ0,263
4
+ homa/torch/Module.py,sha256=UuRTJXxp7hlQHHL2dBGVlHlyA2OFgJ1Xz9SAff8Iv3c,171
5
+ homa/torch/__init__.py,sha256=-pZ00nx4KYkJU3435Jfb-Nmeks6nU4w4lw4AE7La_0A,27
6
+ homa-0.0.11.dist-info/METADATA,sha256=FVcYyqRouigTuo0OO77b3etbPvax_V_Orx8BDu6LqE8,520
7
+ homa-0.0.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
8
+ homa-0.0.11.dist-info/top_level.txt,sha256=tmOfy2tuaAwc3W5-i6j61_vYJsXgR4ivBWkhJ3ZtJDc,5
9
+ homa-0.0.11.dist-info/RECORD,,
@@ -1,7 +0,0 @@
1
- homa/__init__.py,sha256=NBYFKizG8UASiz5HLsEBqzXNGlWr78xm4sLr5hxKvjU,46
2
- homa/device.py,sha256=9kKXfpYfnEk2cFQWPfcJrVloHgC_SSbP4I8IRY9TYk4,343
3
- homa/settings.py,sha256=CPZDPvs1380O7SY7FcSKol8kBVFVVYFgSJl3YEyJuZ0,263
4
- homa-0.0.1.dist-info/METADATA,sha256=TMXUMO31Dt-5DrlD4maXJTNtWYlzMBFJvg3JONUhXj4,366
5
- homa-0.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
6
- homa-0.0.1.dist-info/top_level.txt,sha256=tmOfy2tuaAwc3W5-i6j61_vYJsXgR4ivBWkhJ3ZtJDc,5
7
- homa-0.0.1.dist-info/RECORD,,
File without changes