jax-autovmap

View on PyPIReverse Dependencies (0)

0.2.0 jax_autovmap-0.2.0-py3-none-any.whl
0.1.0 jax_autovmap-0.1.0-py3-none-any.whl

Wheel Details

Project: jax-autovmap
Version: 0.2.0
Filename: jax_autovmap-0.2.0-py3-none-any.whl
Download: [link]
Size: 7055
MD5: 30a23019763cdcf1600a8f337b827ccb
SHA256: 980c54116932fc0edda7512650f0b9efce9074c2a7f2fed6bcf0cdb44e229436
Uploaded: 2024-12-26 03:02:18 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jax_autovmap
Version: 0.2.0
Summary: Automatically broadcast inputs by dynamically applying jax.vmap given un-batched argument dimensions.
Author-Email: Mathis Gerdes <mathisgerdes[at]gmail.com>
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.8
Requires-Dist: jax (>=0.4.1)
Requires-Dist: numpy
Description-Content-Type: text/markdown
[Description omitted; length: 4076 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.6.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
jax_autovmap.py sha256=6zlLd7E4wNg3axjSM1-t1QgGG3CGW7KVsxb0pXxyHLg 8577
test_autovmap.py sha256=tmGdeKkKlabp4CHH4UxWBttJ8gwOmUpeQqe5oRESKyg 3791
jax_autovmap-0.2.0.dist-info/METADATA sha256=ec9l2IZoj5Pk5babetw2cbzDOvo6lqroq8eRurGscQU 4553
jax_autovmap-0.2.0.dist-info/WHEEL sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U 91
jax_autovmap-0.2.0.dist-info/top_level.txt sha256=E6W2kx1BeRs0QEcoQ2-lKMr627RijhB3nOJ113W4AvQ 27
jax_autovmap-0.2.0.dist-info/RECORD

top_level.txt

jax_autovmap
test_autovmap