xbarray 0.0.1a5__tar.gz → 0.0.1a6__tar.gz
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 xbarray might be problematic. Click here for more details.
- {xbarray-0.0.1a5/xbarray.egg-info → xbarray-0.0.1a6}/PKG-INFO +1 -1
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/pyproject.toml +1 -1
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/jax/__init__.py +1 -1
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/numpy/__init__.py +1 -1
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/pytorch/__init__.py +1 -1
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/jax.py +4 -1
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/numpy.py +4 -1
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/pytorch.py +4 -1
- {xbarray-0.0.1a5 → xbarray-0.0.1a6/xbarray.egg-info}/PKG-INFO +1 -1
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/LICENSE +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/README.md +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/__init__.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_2024_12/__init__.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_2024_12/_api_constant.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_2024_12/_api_fft_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_2024_12/_api_linalg_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_2024_12/_api_return_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_2024_12/_api_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_2024_12/_array_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_compat/__init__.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_compat/_api_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_compat/_array_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_extra/__init__.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_extra/_api_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/array_api_typing/typing_extra/_at.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/setup.cfg +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/__init__.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/_common/implementations.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/jax/_extra.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/jax/_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/jax/random.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/numpy/_extra.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/numpy/_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/numpy/random.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/pytorch/_extra.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/pytorch/_typing.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/implementations/pytorch/random.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/serialization/__init__.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/_src/serialization/serialization_map.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/base/__init__.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/base/base.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray/cls_impl/cls_base.py +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray.egg-info/SOURCES.txt +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray.egg-info/dependency_links.txt +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray.egg-info/requires.txt +0 -0
- {xbarray-0.0.1a5 → xbarray-0.0.1a6}/xbarray.egg-info/top_level.txt +0 -0
|
@@ -2,7 +2,10 @@ from xbarray.cls_impl.cls_base import ComputeBackendImplCls
|
|
|
2
2
|
from ._src.implementations import jax as jax_impl
|
|
3
3
|
|
|
4
4
|
class JaxComputeBackend(metaclass=ComputeBackendImplCls):
|
|
5
|
-
|
|
5
|
+
ARRAY_TYPE = jax_impl.ARRAY_TYPE
|
|
6
|
+
DTYPE_TYPE = jax_impl.DTYPE_TYPE
|
|
7
|
+
DEVICE_TYPE = jax_impl.DEVICE_TYPE
|
|
8
|
+
RNG_TYPE = jax_impl.RNG_TYPE
|
|
6
9
|
|
|
7
10
|
for name in dir(jax_impl):
|
|
8
11
|
if not name.startswith('_'):
|
|
@@ -2,7 +2,10 @@ from xbarray.cls_impl.cls_base import ComputeBackendImplCls
|
|
|
2
2
|
from ._src.implementations import numpy as numpy_impl
|
|
3
3
|
|
|
4
4
|
class NumpyComputeBackend(metaclass=ComputeBackendImplCls):
|
|
5
|
-
|
|
5
|
+
ARRAY_TYPE = numpy_impl.ARRAY_TYPE
|
|
6
|
+
DTYPE_TYPE = numpy_impl.DTYPE_TYPE
|
|
7
|
+
DEVICE_TYPE = numpy_impl.DEVICE_TYPE
|
|
8
|
+
RNG_TYPE = numpy_impl.RNG_TYPE
|
|
6
9
|
|
|
7
10
|
for name in dir(numpy_impl):
|
|
8
11
|
if not name.startswith('_'):
|
|
@@ -2,7 +2,10 @@ from xbarray.cls_impl.cls_base import ComputeBackendImplCls
|
|
|
2
2
|
from ._src.implementations import pytorch as pytorch_impl
|
|
3
3
|
|
|
4
4
|
class PytorchComputeBackend(metaclass=ComputeBackendImplCls):
|
|
5
|
-
|
|
5
|
+
ARRAY_TYPE = pytorch_impl.ARRAY_TYPE
|
|
6
|
+
DTYPE_TYPE = pytorch_impl.DTYPE_TYPE
|
|
7
|
+
DEVICE_TYPE = pytorch_impl.DEVICE_TYPE
|
|
8
|
+
RNG_TYPE = pytorch_impl.RNG_TYPE
|
|
6
9
|
|
|
7
10
|
for name in dir(pytorch_impl):
|
|
8
11
|
if not name.startswith('_'):
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|