torch-bessel 0.0.2__cp312-cp312-macosx_11_0_arm64.whl → 0.0.3__cp312-cp312-macosx_11_0_arm64.whl

{torch_bessel-0.0.2.dist-info → torch_bessel-0.0.3.dist-info}/METADATA

@@ -1,8 +1,8 @@
 Metadata-Version: 2.2
 Name: torch_bessel
-Version: 0.0.2
+Version: 0.0.3
 Summary: PyTorch extension package for Bessel functions with arbitrary real order and complex inputs
-Home-page: https://github.com/pytorch/torch-bessel
+Home-page: https://github.com/hchau630/torch-bessel
 Author: Ho Yin Chau
 Requires-Python: >= 3.9
 Description-Content-Type: text/markdown
@@ -20,20 +20,22 @@ Dynamic: summary
 PyTorch extension package for Bessel functions with arbitrary real order and complex inputs
 
 # Install
+Currently only supports Linux (with CUDA 12.4) or MacOS (Apple silicon, cpu only) with python >= 3.9, <= 3.12.
 ```
 pip install torch-bessel
 ```
 
-# Example usage
+# Example
 ```
 import torch_bessel
 
-z = torch.randn(10) + 1j
+real, imag = torch.randn(2, 5, device="cuda")
+z = torch.complex(real.abs(), imag)  # inputs on the left-half complex plane are set to NaNs.
 torch_bessel.ops.modified_bessel_k0(z)
 ```
 
 # Implemented functions
-- `modified_bessel_k0`: Same as `torch.special.modified_bessel_k0`, but also handles backpropagation and complex inputs with $\mathrm{Im}(z) \geq 0$ on cpu and cuda.
+- `modified_bessel_k0`: Same as `torch.special.modified_bessel_k0`, but also handles backpropagation and complex inputs with $\mathrm{Re}(z) \geq 0$ on cpu and cuda.
 
 # WIP
 - `modified_bessel_kv`: Analogue of `scipy.special.kv`.

torch_bessel-0.0.3.dist-info/RECORD

@@ -0,0 +1,8 @@
+torch_bessel/_C.cpython-312-darwin.so,sha256=d2qpLB8simaybQxmgXOrrHwvsG3yw32yoehTIX5PsO0,232280
+torch_bessel/__init__.py,sha256=oohbWz8vxekl7kqDNSWqDH3ORabf9-Tug1KJryKw51A,230
+torch_bessel/ops.py,sha256=Q9BrLxi15MS53xSt_S9dyE3g8_8_GCFYhfAztIor8Fw,2043
+torch_bessel-0.0.3.dist-info/LICENSE,sha256=do0DI6wu4mF3VXnEXXPYZqVEatoRSSamgz9t80wU7_o,1068
+torch_bessel-0.0.3.dist-info/METADATA,sha256=7hH6thd8bu4oaKYfNq5v6V4JkUe6g6_C9Irl618y3-8,1220
+torch_bessel-0.0.3.dist-info/WHEEL,sha256=VujM3ypTCyUW6hcTDdK2ej0ARVMxlU1Djlh_zWnDgqk,109
+torch_bessel-0.0.3.dist-info/top_level.txt,sha256=cbDIjTj71LuAlVyyYyDt8fOAeLaVeX3Vums5F2FBa-4,13
+torch_bessel-0.0.3.dist-info/RECORD,,

torch_bessel-0.0.2.dist-info/RECORD

@@ -1,8 +0,0 @@
-torch_bessel/_C.cpython-312-darwin.so,sha256=dwRnSeElZ4yGUJB7oeGLG8HxScqA1QDD6oPYfFHvTpM,232696
-torch_bessel/__init__.py,sha256=oohbWz8vxekl7kqDNSWqDH3ORabf9-Tug1KJryKw51A,230
-torch_bessel/ops.py,sha256=Q9BrLxi15MS53xSt_S9dyE3g8_8_GCFYhfAztIor8Fw,2043
-torch_bessel-0.0.2.dist-info/LICENSE,sha256=do0DI6wu4mF3VXnEXXPYZqVEatoRSSamgz9t80wU7_o,1068
-torch_bessel-0.0.2.dist-info/METADATA,sha256=F8Mbw3qz2XuV82NoMc7HWKQ2lKivcx1X9ht2AScvSfo,999
-torch_bessel-0.0.2.dist-info/WHEEL,sha256=VujM3ypTCyUW6hcTDdK2ej0ARVMxlU1Djlh_zWnDgqk,109
-torch_bessel-0.0.2.dist-info/top_level.txt,sha256=cbDIjTj71LuAlVyyYyDt8fOAeLaVeX3Vums5F2FBa-4,13
-torch_bessel-0.0.2.dist-info/RECORD,,