xbarray 0.0.1a13__py3-none-any.whl

xbarray/transformations/rotation_conversions/jax.py

@@ -0,0 +1,41 @@
+from . import base as base_impl
+from functools import partial
+from xbarray.backends.jax import JaxComputeBackend as BindingBackend
+
+__all__ = [
+    "quaternion_to_matrix",
+    "matrix_to_quaternion",
+    "euler_angles_to_matrix",
+    "matrix_to_euler_angles",
+    "random_quaternions",
+    "random_rotations",
+    "random_rotation",
+    "standardize_quaternion",
+    "quaternion_multiply",
+    "quaternion_invert",
+    "quaternion_apply",
+    "axis_angle_to_matrix",
+    "matrix_to_axis_angle",
+    "axis_angle_to_quaternion",
+    "quaternion_to_axis_angle",
+    "rotation_6d_to_matrix",
+    "matrix_to_rotation_6d",
+]
+
+quaternion_to_matrix = partial(base_impl.quaternion_to_matrix, BindingBackend)
+matrix_to_quaternion = partial(base_impl.matrix_to_quaternion, BindingBackend)
+euler_angles_to_matrix = partial(base_impl.euler_angles_to_matrix, BindingBackend)
+matrix_to_euler_angles = partial(base_impl.matrix_to_euler_angles, BindingBackend)
+random_quaternions = partial(base_impl.random_quaternions, BindingBackend)
+random_rotations = partial(base_impl.random_rotations, BindingBackend)
+random_rotation = partial(base_impl.random_rotation, BindingBackend)
+standardize_quaternion = partial(base_impl.standardize_quaternion, BindingBackend)
+quaternion_multiply = partial(base_impl.quaternion_multiply, BindingBackend)
+quaternion_invert = partial(base_impl.quaternion_invert, BindingBackend)
+quaternion_apply = partial(base_impl.quaternion_apply, BindingBackend)
+axis_angle_to_matrix = partial(base_impl.axis_angle_to_matrix, BindingBackend)
+matrix_to_axis_angle = partial(base_impl.matrix_to_axis_angle, BindingBackend)
+axis_angle_to_quaternion = partial(base_impl.axis_angle_to_quaternion, BindingBackend)
+quaternion_to_axis_angle = partial(base_impl.quaternion_to_axis_angle, BindingBackend)
+rotation_6d_to_matrix = partial(base_impl.rotation_6d_to_matrix, BindingBackend)
+matrix_to_rotation_6d = partial(base_impl.matrix_to_rotation_6d, BindingBackend)

xbarray/transformations/rotation_conversions/numpy.py

@@ -0,0 +1,41 @@
+from . import base as base_impl
+from functools import partial
+from xbarray.backends.numpy import NumpyComputeBackend as BindingBackend
+
+__all__ = [
+    "quaternion_to_matrix",
+    "matrix_to_quaternion",
+    "euler_angles_to_matrix",
+    "matrix_to_euler_angles",
+    "random_quaternions",
+    "random_rotations",
+    "random_rotation",
+    "standardize_quaternion",
+    "quaternion_multiply",
+    "quaternion_invert",
+    "quaternion_apply",
+    "axis_angle_to_matrix",
+    "matrix_to_axis_angle",
+    "axis_angle_to_quaternion",
+    "quaternion_to_axis_angle",
+    "rotation_6d_to_matrix",
+    "matrix_to_rotation_6d",
+]
+
+quaternion_to_matrix = partial(base_impl.quaternion_to_matrix, BindingBackend)
+matrix_to_quaternion = partial(base_impl.matrix_to_quaternion, BindingBackend)
+euler_angles_to_matrix = partial(base_impl.euler_angles_to_matrix, BindingBackend)
+matrix_to_euler_angles = partial(base_impl.matrix_to_euler_angles, BindingBackend)
+random_quaternions = partial(base_impl.random_quaternions, BindingBackend)
+random_rotations = partial(base_impl.random_rotations, BindingBackend)
+random_rotation = partial(base_impl.random_rotation, BindingBackend)
+standardize_quaternion = partial(base_impl.standardize_quaternion, BindingBackend)
+quaternion_multiply = partial(base_impl.quaternion_multiply, BindingBackend)
+quaternion_invert = partial(base_impl.quaternion_invert, BindingBackend)
+quaternion_apply = partial(base_impl.quaternion_apply, BindingBackend)
+axis_angle_to_matrix = partial(base_impl.axis_angle_to_matrix, BindingBackend)
+matrix_to_axis_angle = partial(base_impl.matrix_to_axis_angle, BindingBackend)
+axis_angle_to_quaternion = partial(base_impl.axis_angle_to_quaternion, BindingBackend)
+quaternion_to_axis_angle = partial(base_impl.quaternion_to_axis_angle, BindingBackend)
+rotation_6d_to_matrix = partial(base_impl.rotation_6d_to_matrix, BindingBackend)
+matrix_to_rotation_6d = partial(base_impl.matrix_to_rotation_6d, BindingBackend)

xbarray/transformations/rotation_conversions/pytorch.py

@@ -0,0 +1,41 @@
+from . import base as base_impl
+from functools import partial
+from xbarray.backends.pytorch import PytorchComputeBackend as BindingBackend
+
+__all__ = [
+    "quaternion_to_matrix",
+    "matrix_to_quaternion",
+    "euler_angles_to_matrix",
+    "matrix_to_euler_angles",
+    "random_quaternions",
+    "random_rotations",
+    "random_rotation",
+    "standardize_quaternion",
+    "quaternion_multiply",
+    "quaternion_invert",
+    "quaternion_apply",
+    "axis_angle_to_matrix",
+    "matrix_to_axis_angle",
+    "axis_angle_to_quaternion",
+    "quaternion_to_axis_angle",
+    "rotation_6d_to_matrix",
+    "matrix_to_rotation_6d",
+]
+
+quaternion_to_matrix = partial(base_impl.quaternion_to_matrix, BindingBackend)
+matrix_to_quaternion = partial(base_impl.matrix_to_quaternion, BindingBackend)
+euler_angles_to_matrix = partial(base_impl.euler_angles_to_matrix, BindingBackend)
+matrix_to_euler_angles = partial(base_impl.matrix_to_euler_angles, BindingBackend)
+random_quaternions = partial(base_impl.random_quaternions, BindingBackend)
+random_rotations = partial(base_impl.random_rotations, BindingBackend)
+random_rotation = partial(base_impl.random_rotation, BindingBackend)
+standardize_quaternion = partial(base_impl.standardize_quaternion, BindingBackend)
+quaternion_multiply = partial(base_impl.quaternion_multiply, BindingBackend)
+quaternion_invert = partial(base_impl.quaternion_invert, BindingBackend)
+quaternion_apply = partial(base_impl.quaternion_apply, BindingBackend)
+axis_angle_to_matrix = partial(base_impl.axis_angle_to_matrix, BindingBackend)
+matrix_to_axis_angle = partial(base_impl.matrix_to_axis_angle, BindingBackend)
+axis_angle_to_quaternion = partial(base_impl.axis_angle_to_quaternion, BindingBackend)
+quaternion_to_axis_angle = partial(base_impl.quaternion_to_axis_angle, BindingBackend)
+rotation_6d_to_matrix = partial(base_impl.rotation_6d_to_matrix, BindingBackend)
+matrix_to_rotation_6d = partial(base_impl.matrix_to_rotation_6d, BindingBackend)

xbarray-0.0.1a13.dist-info/METADATA

@@ -0,0 +1,20 @@
+Metadata-Version: 2.4
+Name: xbarray
+Version: 0.0.1a13
+Summary: Cross-backend Python array library based on the Array API Standard.
+Project-URL: Homepage, https://github.com/UniEnvOrg/XBArray
+Project-URL: Documentation, https://github.com/UniEnvOrg/XBArray
+Project-URL: Repository, https://github.com/UniEnvOrg/XBArray
+Project-URL: Issues, https://github.com/UniEnvOrg/XBArray/issues
+Project-URL: Changelog, https://github.com/UniEnvOrg/XBArray/blob/main/CHANGELOG.md
+Requires-Python: >=3.10
+License-File: LICENSE
+Requires-Dist: typing_extensions>=4.5
+Requires-Dist: array_api_compat
+Requires-Dist: array_api_extra
+Requires-Dist: numpy
+Provides-Extra: torch
+Requires-Dist: torch; extra == "torch"
+Provides-Extra: jax
+Requires-Dist: jax; extra == "jax"
+Dynamic: license-file

xbarray-0.0.1a13.dist-info/RECORD

@@ -0,0 +1,51 @@
+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=rIygF4PAr7G1PK3cVOwQNrev_kGCuMcfIJ9i8L0u1tI,2462
+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=KgXKAKk4obhlDaydQmjePwYqAAj_4Ag4JBJbfcc303M,54
+xbarray/jax.py,sha256=IeMSMi0TQ4A1kho1cyNOXtB0XeaxTKaO_2ymwhW_h3U,174
+xbarray/numpy.py,sha256=5EcCJ8q8AU9r58eZSAJPgA63-RCqjxaWwaHrQp7XN9I,178
+xbarray/pytorch.py,sha256=L_ZLPKKqg8KrqgFILmzkGLtjafec0aza52R3TW4wniw,182
+xbarray/backends/_cls_base.py,sha256=O-99tV_9DxfqFfMNC7ZslPS4tGeIilpN6JGtxqbFCPY,312
+xbarray/backends/base.py,sha256=zoABR4WXHFGyqhiUW8Tnv1PnScwEFxz-G2J86-WBshQ,6768
+xbarray/backends/jax.py,sha256=68Tda5mWVucoviZUzYEqS9V4BELKA1hapKaFuWKr_ZE,633
+xbarray/backends/numpy.py,sha256=Dfdu3fHt8YhU1tdOwd3xmtzdZXOXs0fzxnd2LjaqJXU,663
+xbarray/backends/pytorch.py,sha256=1Dfy5XLxagK3sjRgt57Q8ppE1-dOhF0T2HipLHRoCVg,758
+xbarray/backends/_implementations/_common/implementations.py,sha256=Ra1-wGN_ZBxIFe4tuR-vCHYLgwkeeFoQtKwVEUwn_To,3834
+xbarray/backends/_implementations/jax/__init__.py,sha256=vOtCkwe98vNI1qQOu0Xv9OIaVucaJp6bEI9_mzicBH0,1014
+xbarray/backends/_implementations/jax/_extra.py,sha256=_E1vXolN86-zOdbN60S_YnQ8lePqPwB7l-EKB_T4KNE,3515
+xbarray/backends/_implementations/jax/_typing.py,sha256=U9BUxHNEjFB0LHF1KMrFLbh6E5kVvpAF8bZUbLNf25E,278
+xbarray/backends/_implementations/jax/random.py,sha256=bIyGV0Nc1PF4PaU0F-paOG4bAENm8ngJFr20voMaYsg,3316
+xbarray/backends/_implementations/numpy/__init__.py,sha256=hJEcFR86pb1G72MFS1X15qwI_-BuPxy5SS5PXB0alQI,792
+xbarray/backends/_implementations/numpy/_extra.py,sha256=DEDWkoK-rTKemMa7Lh3b-FSGPsdnfBA6i-6CWiIlV-M,2379
+xbarray/backends/_implementations/numpy/_typing.py,sha256=pgjLLAipwFsIk0QdgrAA4PEvjF_ugHbzfSTbLpA__6o,241
+xbarray/backends/_implementations/numpy/random.py,sha256=C1z2-pMcDMRa7nKhKNTuY5LtTlu9nXvDpH1wY8mdIug,2630
+xbarray/backends/_implementations/pytorch/__init__.py,sha256=s2mF3jWI356oedPZevKYARCFhlXH0hqS9OgBzxMSgHQ,795
+xbarray/backends/_implementations/pytorch/_extra.py,sha256=mVRnxYQg-1pNtUTsb-4YV25Wvhh908awsKKT1B7P_2Y,3638
+xbarray/backends/_implementations/pytorch/_typing.py,sha256=qSnNZD3IpgSoOTl3TBRBv2ifSAHOw0aR9uyzfV5KYVw,204
+xbarray/backends/_implementations/pytorch/random.py,sha256=p4ZDsTc--z5XslVzel-iEVDtoKizrv39_1tfR4-469o,2657
+xbarray/transformations/pointcloud/__init__.py,sha256=GLdMGsm5_CnATyh19At20Hmb1MsXZvPxNvg70bHYChk,19
+xbarray/transformations/pointcloud/base.py,sha256=g63dLmkhzEWOEpKjK3yZe7KW09yQ5YNSA_whDyH6L1A,20413
+xbarray/transformations/pointcloud/jax.py,sha256=GUq6pHt0pn6r8PpY6mpMjGSa7zyT13sePEDup7yo6dI,1081
+xbarray/transformations/pointcloud/numpy.py,sha256=ZwWb0LTmz7jfMmWQqFq--S0jyRyK2f3sySnQCA8EZPw,1084
+xbarray/transformations/pointcloud/pytorch.py,sha256=PVCkB-ZpLu1rieD0Xpt_9jAG6qDtJjN6IviXcMbxnIg,1088
+xbarray/transformations/rotation_conversions/__init__.py,sha256=GLdMGsm5_CnATyh19At20Hmb1MsXZvPxNvg70bHYChk,19
+xbarray/transformations/rotation_conversions/base.py,sha256=L578W9mLgAdGnsnOOU-RYUxSqZLk0VTJdHANNepDAcU,26909
+xbarray/transformations/rotation_conversions/jax.py,sha256=0BmQ_B_eqedqXntWjgNcioNfFuecy4LZNRf347kltsU,1956
+xbarray/transformations/rotation_conversions/numpy.py,sha256=j-44lmMydo8lMhvGLMWMEmUFZlDtEa3VrgxAPLIIbDg,1960
+xbarray/transformations/rotation_conversions/pytorch.py,sha256=Eu2kOolvQqm56NPiVYpuBzQ3MUc1ANREw3J4AftXoxM,1964
+xbarray-0.0.1a13.dist-info/licenses/LICENSE,sha256=6P0HCOancSfch0dNycuDIe8_qwS0Id97Ih_8hjJ2PFI,1067
+xbarray-0.0.1a13.dist-info/METADATA,sha256=1y3gwpjzZ4sFKMDFJnhXyH08ryF5xUWLSMQA1whXC-0,773
+xbarray-0.0.1a13.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+xbarray-0.0.1a13.dist-info/top_level.txt,sha256=VriXuFyU48Du4HQMzROSArhwqB6EZYY0n0mipgUqB9A,25
+xbarray-0.0.1a13.dist-info/RECORD,,

xbarray-0.0.1a13.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.10.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

xbarray-0.0.1a13.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Yunhao Cao
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

xbarray-0.0.1a13.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+array_api_typing
+xbarray