xbarray 0.0.1a1__py3-none-any.whl → 0.0.1a2__py3-none-any.whl

xbarray/__init__.py

@@ -1 +1 @@
-from .base import *
+from .base import * # Import Abstract Typings

xbarray/{jax → _src/implementations/jax}/__init__.py

@@ -9,16 +9,22 @@ else:
     import jax.experimental.array_api as compat_module
     from jax.experimental.array_api import *
 
+from array_api_compat.common._helpers import *
+
 # Import and bind all functions from array_api_extra before exposing them
 import array_api_extra
 from functools import partial
 for api_name in dir(array_api_extra):
     if api_name.startswith('_'):
         continue
-    globals()[api_name] = partial(
-        getattr(array_api_extra, api_name),
-        xp=compat_module
-    )
+
+    if api_name == 'at':
+        globals()[api_name] = getattr(array_api_extra, api_name)
+    else:
+        globals()[api_name] = partial(
+            getattr(array_api_extra, api_name),
+            xp=compat_module
+        )
 
 from ._typing import *
 from ._extra import *

xbarray/{jax → _src/implementations/jax}/_extra.py

@@ -77,7 +77,7 @@ def dtype_is_boolean(
 ) -> bool:
     return dtype == np.bool_ or dtype == bool
 
-from xbarray.common.implementations import *
+from .._common.implementations import *
 if hasattr(jax.numpy, "__array_api_version__"):
     compat_module = jax.numpy
 else:

xbarray/{jax → _src/implementations/jax}/random.py

@@ -26,9 +26,9 @@ def random_number_generator(
 
 def random_discrete_uniform(
     shape : Union[int, Tuple[int, ...]], 
+    /,
     from_num : int, 
     to_num : int, 
-    /,
     *,
     rng : RNG_TYPE, 
     dtype : Optional[DTYPE_TYPE] = None, 
@@ -70,7 +70,6 @@ def random_exponential(
         data = jax.device_put(data, device)
     return new_rng, data
 
-@classmethod
 def random_normal(
     shape: Union[int, Tuple[int, ...]],
     /,

xbarray/{numpy → _src/implementations/numpy}/__init__.py

@@ -9,10 +9,14 @@ from functools import partial
 for api_name in dir(array_api_extra):
     if api_name.startswith('_'):
         continue
-    globals()[api_name] = partial(
-        getattr(array_api_extra, api_name),
-        xp=compat_module
-    )
+
+    if api_name == 'at':
+        globals()[api_name] = getattr(array_api_extra, api_name)
+    else:
+        globals()[api_name] = partial(
+            getattr(array_api_extra, api_name),
+            xp=compat_module
+        )
 
 from ._typing import *
 from ._extra import *

xbarray/{numpy → _src/implementations/numpy}/_extra.py

@@ -68,7 +68,7 @@ def dtype_is_boolean(
 ) -> bool:
     return dtype == np.bool_ or dtype == bool
 
-from xbarray.common.implementations import get_abbreviate_array_function, get_map_fn_over_arrays_function
+from .._common.implementations import get_abbreviate_array_function, get_map_fn_over_arrays_function
 from array_api_compat import numpy as compat_module
 abbreviate_array = get_abbreviate_array_function(
     backend=compat_module,

xbarray/{numpy → _src/implementations/numpy}/random.py

@@ -21,9 +21,9 @@ def random_number_generator(
 
 def random_discrete_uniform(
     shape : Union[int, Tuple[int, ...]], 
+    /,
     from_num : int, 
     to_num : int, 
-    /,
     *,
     rng : RNG_TYPE, 
     dtype : Optional[DTYPE_TYPE] = None, 
@@ -62,7 +62,6 @@ def random_exponential(
         t = t.astype(dtype)
     return rng, t
 
-@classmethod
 def random_normal(
     shape: Union[int, Tuple[int, ...]],
     /,

xbarray/{pytorch → _src/implementations/pytorch}/__init__.py

@@ -10,10 +10,14 @@ from functools import partial
 for api_name in dir(array_api_extra):
     if api_name.startswith('_'):
         continue
-    globals()[api_name] = partial(
-        getattr(array_api_extra, api_name),
-        xp=compat_module
-    )
+
+    if api_name == 'at':
+        globals()[api_name] = getattr(array_api_extra, api_name)
+    else:
+        globals()[api_name] = partial(
+            getattr(array_api_extra, api_name),
+            xp=compat_module
+        )
 
 from ._typing import *
 from ._extra import *

xbarray/{pytorch → _src/implementations/pytorch}/_extra.py

@@ -95,7 +95,7 @@ def dtype_is_boolean(
 ) -> bool:
     return dtype == torch.bool
 
-from xbarray.common.implementations import get_abbreviate_array_function, get_map_fn_over_arrays_function
+from .._common.implementations import get_abbreviate_array_function, get_map_fn_over_arrays_function
 from array_api_compat import torch as compat_module
 abbreviate_array = get_abbreviate_array_function(
     compat_module, 

xbarray/{pytorch → _src/implementations/pytorch}/random.py

@@ -27,9 +27,9 @@ def random_number_generator(
 
 def random_discrete_uniform(
     shape : Union[int, Tuple[int, ...]], 
+    /,
     from_num : int, 
     to_num : int, 
-    /,
     *,
     rng : RNG_TYPE, 
     dtype : Optional[DTYPE_TYPE] = None, 
@@ -64,7 +64,6 @@ def random_exponential(
     t = t.exponential_(lambd, generator=rng)
     return rng, t
 
-@classmethod
 def random_normal(
     shape: Union[int, Tuple[int, ...]],
     /,

xbarray/_src/serialization/__init__.py

@@ -0,0 +1 @@
+from .serialization_map import *

xbarray/_src/serialization/serialization_map.py

@@ -0,0 +1,31 @@
+from typing import Type, Dict, Any, Optional, List, Callable
+from types import ModuleType
+import importlib
+
+__all__ = [
+    "implementation_module_to_name",
+    "name_to_implementation_module",
+]
+
+def implementation_module_to_name(module : ModuleType) -> str:
+    """
+    Convert a backend module to its simplified name.
+    """
+    full_name = module.__name__
+    if not full_name.startswith("xbarray.implementations."):
+        raise ValueError(f"Module {full_name} is not a valid xbarray backend module.")
+
+    submodule_name = full_name[len("xbarray.implementations"):]
+    if '.' in submodule_name:
+        raise ValueError(f"Module {full_name} is not a valid xbarray backend module.")
+    return submodule_name
+
+def name_to_implementation_module(name: str) -> Type[ModuleType]:
+    """
+    Convert a simplified backend name to its module.
+    """
+    try:
+        return importlib.import_module(f"xbarray.implementations.{name}")
+    except ImportError as e:
+        raise ImportError(f"Could not import backend module '{name}'.") from e
+    

xbarray/base/base.py

@@ -40,9 +40,9 @@ class RNGBackend(Protocol[BArrayType, BDeviceType, BDtypeType, BRNGType]):
     def random_discrete_uniform(
         self, 
         shape : Union[int, Tuple[int, ...]], 
-        from_num : int, 
-        to_num : int, 
         /,
+        from_num : int, 
+        to_num : int,
         *,
         rng : BRNGType, 
         dtype : Optional[BDtypeType] = None, 

xbarray/cls_impl/cls_base.py

@@ -0,0 +1,10 @@
+from types import ModuleType
+from typing import Type
+from xbarray._src.serialization import implementation_module_to_name, name_to_implementation_module
+
+class ComputeBackendImplCls(Type):
+    def __str__(self):
+        return self.simplified_name
+    
+    def __repr__(self):
+        return self.simplified_name

xbarray/jax.py

@@ -0,0 +1,13 @@
+from xbarray.cls_impl.cls_base import ComputeBackendImplCls
+from ._src.implementations import jax as jax_impl
+
+class JaxComputeBackend(metaclass=ComputeBackendImplCls):
+    pass
+
+for name in dir(jax_impl):
+    if not name.startswith('_'):
+        setattr(JaxComputeBackend, name, getattr(jax_impl, name))
+
+__all__ = [
+    'JaxComputeBackend',
+]

xbarray/numpy.py

@@ -0,0 +1,13 @@
+from xbarray.cls_impl.cls_base import ComputeBackendImplCls
+from ._src.implementations import numpy as numpy_impl
+
+class NumpyComputeBackend(metaclass=ComputeBackendImplCls):
+    pass
+
+for name in dir(numpy_impl):
+    if not name.startswith('_'):
+        setattr(NumpyComputeBackend, name, getattr(numpy_impl, name))
+
+__all__ = [
+    'NumpyComputeBackend',
+]

xbarray/pytorch.py

@@ -0,0 +1,13 @@
+from xbarray.cls_impl.cls_base import ComputeBackendImplCls
+from ._src.implementations import pytorch as pytorch_impl
+
+class PytorchComputeBackend(metaclass=ComputeBackendImplCls):
+    pass
+
+for name in dir(pytorch_impl):
+    if not name.startswith('_'):
+        setattr(PytorchComputeBackend, name, getattr(pytorch_impl, name))
+
+__all__ = [
+    'PytorchComputeBackend',
+]

{xbarray-0.0.1a1.dist-info → xbarray-0.0.1a2.dist-info}/METADATA

@@ -1,6 +1,7 @@
 Metadata-Version: 2.4
 Name: xbarray
-Version: 0.0.1a1
+Version: 0.0.1a2
+Summary: Cross-backend Python array library based on the Array API Standard.
 Requires-Python: >=3.10
 License-File: LICENSE
 Requires-Dist: typing_extensions>=4.5

xbarray-0.0.1a2.dist-info/RECORD

@@ -0,0 +1,41 @@
+array_api_typing/__init__.py,sha256=5vcE63PsZ_utc7j2VmLK5evUhar-942p95HyjTiLOc0,179
+array_api_typing/typing_2024_12/__init__.py,sha256=JuD2yojHl3eQI74U75tFhSX5rSy32TR5nKyi714ZfX4,347
+array_api_typing/typing_2024_12/_api_constant.py,sha256=tahdzdi7vAZ3UnM8oDcUqvx2FFsMTAyUbuwgarU6K9U,676
+array_api_typing/typing_2024_12/_api_fft_typing.py,sha256=i48S-9trw3s72xiCJmeCNGlJLwwOVq0kJC4lD9se_co,38641
+array_api_typing/typing_2024_12/_api_linalg_typing.py,sha256=Dt5fDTwYb-GAmP7duajwPqH3zaJmwvxDEOOzdl5X0LE,48950
+array_api_typing/typing_2024_12/_api_return_typing.py,sha256=7TxSBRv0H0uQ9VnixbyuEXJQnl1uESo102RsAloI9SA,2302
+array_api_typing/typing_2024_12/_api_typing.py,sha256=rFuF7E6b2RrijzQBrZx6_dYEn9Yslq8FqI31L1Ccrf4,294255
+array_api_typing/typing_2024_12/_array_typing.py,sha256=GMqs5LgshujRIh7i0mGtZy4Vt56FSXqBmNlBGidFRnc,57403
+array_api_typing/typing_compat/__init__.py,sha256=JuD2yojHl3eQI74U75tFhSX5rSy32TR5nKyi714ZfX4,347
+array_api_typing/typing_compat/_api_typing.py,sha256=RnGZFD8AEIqOCurcI68xyw7y9W9hm4yWUGf7NlJkNk0,930
+array_api_typing/typing_compat/_array_typing.py,sha256=CVv91wDjjG-8kSofkqgA85qpM-3x34Zh_SNuBFegW9o,1182
+array_api_typing/typing_extra/__init__.py,sha256=YfdhD-Sfk3SCfI9lHmA-PbVLzms1OFF5x0ekzI3aafk,323
+array_api_typing/typing_extra/_api_typing.py,sha256=Jj_E61r35EgecWmBvAzpASV4qub5aQI_O4aL-ngEvQ8,23028
+array_api_typing/typing_extra/_at.py,sha256=S7_YjOwR3a8olZWgwpLDFEfnekRufRqrtfiMLwrWjgo,2202
+xbarray/__init__.py,sha256=k4Ipp7IoODqHWZ-eeBhcU7Ch8FudF8rag6KbeIurrgY,45
+xbarray/jax.py,sha256=3DHAFsWvfurCI0H6loqx4LtmBov9KbIxeHGmwLKrdh4,344
+xbarray/numpy.py,sha256=8thyuJZlm0MJXGx6YWlgxV9ASG-In9jn-pwUFFk7iB4,358
+xbarray/pytorch.py,sha256=NEKlF_QoiOQl0T-aP_CkUxJ3HgFJEm0hmLy9v5I7qAo,372
+xbarray/_src/implementations/_common/implementations.py,sha256=FyZaQZcVdSnH6CUA1LVbUJAK-EAlkz2fDfOOuVyku6Y,2655
+xbarray/_src/implementations/jax/__init__.py,sha256=pkQASOOePuTel32zaSmsNIc9FqL_8Ix0P4NDXMqhupE,827
+xbarray/_src/implementations/jax/_extra.py,sha256=6bk789VD17LJUE4jmV2E_xosgSqpVo2W2cmySBckrQ8,2499
+xbarray/_src/implementations/jax/_typing.py,sha256=U9BUxHNEjFB0LHF1KMrFLbh6E5kVvpAF8bZUbLNf25E,278
+xbarray/_src/implementations/jax/random.py,sha256=k6vuNLTOSUClMKI6My0-Xekkeu6hJrUyNib1TziuGag,3312
+xbarray/_src/implementations/numpy/__init__.py,sha256=s_tC8UT0ULs9tpZ5JFiEIOnv5sRAVBZFM5Wh90T44Wc,640
+xbarray/_src/implementations/numpy/_extra.py,sha256=J6mPhMri9sI9UK52Sc61IHM4uSLZ3eq9SM-yKFwOwJQ,1977
+xbarray/_src/implementations/numpy/_typing.py,sha256=pgjLLAipwFsIk0QdgrAA4PEvjF_ugHbzfSTbLpA__6o,241
+xbarray/_src/implementations/numpy/random.py,sha256=C1z2-pMcDMRa7nKhKNTuY5LtTlu9nXvDpH1wY8mdIug,2630
+xbarray/_src/implementations/pytorch/__init__.py,sha256=CIXK76WnheuEkVjdU_efCgLxBx3ABhcTdbaImRzmkNM,643
+xbarray/_src/implementations/pytorch/_extra.py,sha256=mqA2JlyVhRJrBegGdkV5kCkb9I6L6bz_Iy7NCd_xcr8,2862
+xbarray/_src/implementations/pytorch/_typing.py,sha256=qSnNZD3IpgSoOTl3TBRBv2ifSAHOw0aR9uyzfV5KYVw,204
+xbarray/_src/implementations/pytorch/random.py,sha256=p4ZDsTc--z5XslVzel-iEVDtoKizrv39_1tfR4-469o,2657
+xbarray/_src/serialization/__init__.py,sha256=xnfYis9UhhvrCGtzzo7dRLb22MEpaNPAJrpljjP1YUU,32
+xbarray/_src/serialization/serialization_map.py,sha256=05zfqip8qPCt-LgM_cx_zLc4y0I4DoUT7_rbU4seNhw,1048
+xbarray/base/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
+xbarray/base/base.py,sha256=V1yAXu2wUu6q50okX7JQ8onkGhHIQ0lkPkYOolwc45A,5928
+xbarray/cls_impl/cls_base.py,sha256=MUvJeMm4UVW4jXwfVP02GiCzqftrAND-JYvThge4PUw,312
+xbarray-0.0.1a2.dist-info/licenses/LICENSE,sha256=6P0HCOancSfch0dNycuDIe8_qwS0Id97Ih_8hjJ2PFI,1067
+xbarray-0.0.1a2.dist-info/METADATA,sha256=MNcr-e1byBuMtn2Z-ouYk5NPXMqdyMNQG5LbWDIHt0s,436
+xbarray-0.0.1a2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xbarray-0.0.1a2.dist-info/top_level.txt,sha256=VriXuFyU48Du4HQMzROSArhwqB6EZYY0n0mipgUqB9A,25
+xbarray-0.0.1a2.dist-info/RECORD,,

xbarray-0.0.1a1.dist-info/RECORD

@@ -1,35 +0,0 @@
-array_api_typing/__init__.py,sha256=5vcE63PsZ_utc7j2VmLK5evUhar-942p95HyjTiLOc0,179
-array_api_typing/typing_2024_12/__init__.py,sha256=JuD2yojHl3eQI74U75tFhSX5rSy32TR5nKyi714ZfX4,347
-array_api_typing/typing_2024_12/_api_constant.py,sha256=tahdzdi7vAZ3UnM8oDcUqvx2FFsMTAyUbuwgarU6K9U,676
-array_api_typing/typing_2024_12/_api_fft_typing.py,sha256=i48S-9trw3s72xiCJmeCNGlJLwwOVq0kJC4lD9se_co,38641
-array_api_typing/typing_2024_12/_api_linalg_typing.py,sha256=Dt5fDTwYb-GAmP7duajwPqH3zaJmwvxDEOOzdl5X0LE,48950
-array_api_typing/typing_2024_12/_api_return_typing.py,sha256=7TxSBRv0H0uQ9VnixbyuEXJQnl1uESo102RsAloI9SA,2302
-array_api_typing/typing_2024_12/_api_typing.py,sha256=rFuF7E6b2RrijzQBrZx6_dYEn9Yslq8FqI31L1Ccrf4,294255
-array_api_typing/typing_2024_12/_array_typing.py,sha256=GMqs5LgshujRIh7i0mGtZy4Vt56FSXqBmNlBGidFRnc,57403
-array_api_typing/typing_compat/__init__.py,sha256=JuD2yojHl3eQI74U75tFhSX5rSy32TR5nKyi714ZfX4,347
-array_api_typing/typing_compat/_api_typing.py,sha256=RnGZFD8AEIqOCurcI68xyw7y9W9hm4yWUGf7NlJkNk0,930
-array_api_typing/typing_compat/_array_typing.py,sha256=CVv91wDjjG-8kSofkqgA85qpM-3x34Zh_SNuBFegW9o,1182
-array_api_typing/typing_extra/__init__.py,sha256=YfdhD-Sfk3SCfI9lHmA-PbVLzms1OFF5x0ekzI3aafk,323
-array_api_typing/typing_extra/_api_typing.py,sha256=Jj_E61r35EgecWmBvAzpASV4qub5aQI_O4aL-ngEvQ8,23028
-array_api_typing/typing_extra/_at.py,sha256=S7_YjOwR3a8olZWgwpLDFEfnekRufRqrtfiMLwrWjgo,2202
-xbarray/__init__.py,sha256=GLdMGsm5_CnATyh19At20Hmb1MsXZvPxNvg70bHYChk,19
-xbarray/base/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
-xbarray/base/base.py,sha256=hqlizpZOJua9xJfPfc-RR6IAadp-OLPVnBO5cbVVHKQ,5929
-xbarray/common/implementations.py,sha256=FyZaQZcVdSnH6CUA1LVbUJAK-EAlkz2fDfOOuVyku6Y,2655
-xbarray/jax/__init__.py,sha256=J7yN6ytn1fm2qKHZVm9OTt30ItcKMnKnuTxbZgnGKaE,662
-xbarray/jax/_extra.py,sha256=JHZBaGZUXPLCRFdaDBXUjwGD-D4DlLYI0NsxAlGs3FE,2504
-xbarray/jax/_typing.py,sha256=U9BUxHNEjFB0LHF1KMrFLbh6E5kVvpAF8bZUbLNf25E,278
-xbarray/jax/random.py,sha256=owIxqs67MOMXaMN8uwmCu9aLd1Yss24sl9rib4R5Qec,3325
-xbarray/numpy/__init__.py,sha256=trZhNZYIHe0QkQYRB1ZJW-OlteYf_9RVUR3yks6S2iI,523
-xbarray/numpy/_extra.py,sha256=wbzpcVL4UV5TVQlmO73sUwWcBgMiVXFvfCLlCuiB0Uk,1982
-xbarray/numpy/_typing.py,sha256=pgjLLAipwFsIk0QdgrAA4PEvjF_ugHbzfSTbLpA__6o,241
-xbarray/numpy/random.py,sha256=8BaxX_cpzvF7iG0yR0ws3IbG14OJPsX60Gv-1AStplA,2643
-xbarray/pytorch/__init__.py,sha256=wfUkSyrJpDlO2Zggza7HM5SCdYKUBfySO3K71itOHkA,526
-xbarray/pytorch/_extra.py,sha256=VawcjDe0dxiIzvTjIncGxwPPoTw7hWZAPOlZSdDSKl0,2867
-xbarray/pytorch/_typing.py,sha256=qSnNZD3IpgSoOTl3TBRBv2ifSAHOw0aR9uyzfV5KYVw,204
-xbarray/pytorch/random.py,sha256=OJ-YvS2owX5LWdsIzXW-N2YqPhN84Rroco7O5qIMmrg,2670
-xbarray-0.0.1a1.dist-info/licenses/LICENSE,sha256=6P0HCOancSfch0dNycuDIe8_qwS0Id97Ih_8hjJ2PFI,1067
-xbarray-0.0.1a1.dist-info/METADATA,sha256=u2-YoX051dU1Po1mFspLSuIEMmlPdhlWOlNwvbp-8PE,359
-xbarray-0.0.1a1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xbarray-0.0.1a1.dist-info/top_level.txt,sha256=VriXuFyU48Du4HQMzROSArhwqB6EZYY0n0mipgUqB9A,25
-xbarray-0.0.1a1.dist-info/RECORD,,