noxjax

View on PyPIReverse Dependencies (1)

0.4.0 noxjax-0.4.0-py3-none-any.whl

Wheel Details

Project: noxjax
Version: 0.4.0
Filename: noxjax-0.4.0-py3-none-any.whl
Download: [link]
Size: 15651
MD5: 90ebd9e11e6d41e68c419977e447f0ad
SHA256: 0ad3b003ad34dfb0a9e22c08f4ec597c6c191dd53a8b13f4ccc9b6d7c220c0f7
Uploaded: 2024-09-25 00:15:32 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: noxjax
Version: 0.4.0
Summary: Flexible Neural Networks in JAX
Author-Email: Paul Wollenhaupt <paul.wollenhaupt[at]gmail.com>
Project-Url: Homepage, https://github.com/pwolle/flarejax
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Requires-Python: >=3.11
Requires-Dist: jax (>=0.4.33)
Requires-Dist: jaxlib (>=0.4.33)
Requires-Dist: jaxtyping (==0.2.34)
Requires-Dist: beartype (==0.18.5)
Description-Content-Type: text/markdown
[Description omitted; length: 11 characters]

WHEEL

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

RECORD

Path Digest Size
flarejax/__init__.py sha256=gBdwaisWugsz_J7jYmroVkxQcmSghNFq1EJSptqJeUI 316
flarejax/_filter.py sha256=E45lnrFD-uVPaa0oB6O4YgcljhvxE58bMquvFEok_nQ 1261
flarejax/_module.py sha256=OXPTkM7uKufe64HoVZmsmTvQrFFA8yfs5BtTk_81lJw 11784
flarejax/_serial.py sha256=zgcUh-kiYeBCwDK92lRgwX3_Th-wGdaNavmo3NO41h0 4642
flarejax/_tcheck.py sha256=TuVQCez7C-g2zsEB5djs9D7mwaLIW7cF8xEJ8WUrHEI 150
flarejax/net/__init__.py sha256=P-_Q6XvReZIHtHTSiVkbGvtKBjh2GxrZuurQ5ldmm6s 450
flarejax/net/_attention.py sha256=p9Grgm6mXQKpFeteaKdzbCPs8ANHpDx8I_ZAhIOzW0o 6592
flarejax/net/_combine.py sha256=Hfe3iU6B7vRFd9cjVYSx911rvv0dHy4NvP7PcBJeyjE 2911
flarejax/net/_linear.py sha256=Nz9Pga74SuMb98eHW9kfe8m06QKPO1z0V-W_SMa23ds 4723
flarejax/net/_norm.py sha256=TyYlo4fLHryAeWStP1p6YCzI1KNy_nBcXn8c0yM-y28 1928
flarejax/net/_transform.py sha256=I09uKK_PY0AkxaIGRu4leh-16mtojcgMYcYI3HqmFKo 1586
flarejax/opt/__init__.py sha256=wZvg6vCTOpKxdClV-x1Dq4NM2gUGTPJyaKFEt6ztf3k 129
flarejax/opt/_adam.py sha256=O-_8-s66dyESfJ3GXWYwC-Kx-atqkj8fFCVU0G9yn7c 1413
flarejax/opt/_opt.py sha256=UNnGRUBzizmVHiSomDnzFE5xiyupFwj0dR1tbT99hxE 3890
flarejax/opt/_sgd.py sha256=jZNJII5kgjjbpRP3r7ZskyyUX4VgD3OxDVAnPdstsak 329
noxjax-0.4.0.dist-info/METADATA sha256=_c2Pg1fVylsdAd3Y2Ta7iThZi_2plAPvR8DO671OfNs 502
noxjax-0.4.0.dist-info/WHEEL sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ 91
noxjax-0.4.0.dist-info/top_level.txt sha256=in8KL9oiIhp1ilxwmPPd27j-Q0aVXzIf1BEi8sW6tR4 9
noxjax-0.4.0.dist-info/RECORD

top_level.txt

flarejax