nnx

View on PyPIReverse Dependencies (0)

0.0.8 nnx-0.0.8-py3-none-any.whl

Wheel Details

Project: nnx
Version: 0.0.8
Filename: nnx-0.0.8-py3-none-any.whl
Download: [link]
Size: 49835
MD5: 88407e5dfd4844a215a55a7b91c9a407
SHA256: b140efed7809992b3a38da0a0c70bf90a06d5505ea729f603babffd1e0881e70
Uploaded: 2023-07-12 16:44:26 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: nnx
Version: 0.0.8
Author: Cristian Garcia
Author-Email: cgarcia.e88[at]gmail.com
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Python: >=3.9,<3.12
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: optax
Requires-Dist: typing-extensions
Description-Content-Type: text/markdown
[Description omitted; length: 21251 characters]

WHEEL

Wheel-Version: 1.0
Generator: poetry-core 1.5.1
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
nnx/__init__.py sha256=Lq9KV5eS3U00FmUPqBOqGVZKz9jGpmblnLGYYmMSF1E 1410
nnx/containers.py sha256=gCANOSSm2ZafqJUUXzCIPRH6r-danHoSdP68nCg_K6Q 5241
nnx/contextlib.py sha256=QimkIGslT7JHhVcYUU8gpxaExizCvlSvnN6pEO0gmqU 5417
nnx/dataclasses.py sha256=fNj5oWpKdUnk0DvuFAgfFP8UoEwe21FzXr7DoKahfcw 4512
nnx/errors.py sha256=fM_BHImRy8CYnqxX1PL_cBobnF7VdhMT6qGMmpQRaQg 45
nnx/helpers.py sha256=-E32hY1diL7bfhBVvYv7xkaeQ-5kNjzfECAZyLReEx0 4345
nnx/ids.py sha256=pPs_J2S0OSRKlho0L8IX5nbQTBhb2sqFQnw9VRX9Rrw 1832
nnx/module.py sha256=OMSmp4QZG5I_QhBcaGH154F9R5GL8CLYWt70znmLcJ4 26625
nnx/nn/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
nnx/nn/activations.py sha256=tFhxZ7Ss3N7fbYeY-jd6iuHP8JsOzWZcEpffvbtmunQ 763
nnx/nn/dtypes.py sha256=LhYy5ZastdjvBERGiHQFfoU7uFclgyu74ttPY-cnWd4 2763
nnx/nn/initializers.py sha256=zmixv3O_i2tkbXMQs39mQmt_8DLnZn3lYPFKpq0iaIg 2147
nnx/nn/linear.py sha256=LlMm-ushgljqdSjOAINklPvQmZZ9xZvwktaqLaEF4R0 16022
nnx/nn/normalization.py sha256=FHyvlBsawxrQaCLydjuGkSfryAJOc2WTAidSynEfB1Y 13511
nnx/nn/stochastic.py sha256=nsf_g6V3egu4LQAn9jXIAUU1WOrOic2LEU5LxHgbo44 2281
nnx/nodes.py sha256=MGi82TKQOWQIAJ6XYsQN3WPiv1b6rR26a_wOKK-U7So 348
nnx/partitioning.py sha256=P6_h3YWTLARoWhre2C_8urR7h-nlwStOzHm9-t34UuM 2037
nnx/pytreelib.py sha256=mNkfwLPYeZ6daaO7_BYVyXhRUxpg3kLbc-2qieh8vOc 8738
nnx/reprlib.py sha256=_m8ARsdXmn9XbuVo9PLJeWEBsvfiyYKw8EIPoMTDN5o 2313
nnx/spmd.py sha256=HzMdfluFTnPXBh5YISwctefLzTOJq2QWcbtJh3VsLmw 13718
nnx/state.py sha256=VTu-Dt0X2xd9dsODsAXpPoMSqV4zj-azYXBofPZxh-8 5103
nnx/tracers.py sha256=sO4KpRK11gKI464qJN88xBYkbufQb8cHpwPx8NFazU4 2412
nnx/transforms.py sha256=6SR23eQjYJggFFMqmfODmMzrxQbZDW3D4MNiA_5nFko 26223
nnx-0.0.8.dist-info/LICENSE sha256=nEeqZTgRHhGmbWc6BWQ5ZWubB-K7KxcGA4mU4Vwc98c 1072
nnx-0.0.8.dist-info/METADATA sha256=9-NVF6t9_-pvk0EThEhvbbJgUoEFA8oB4DjQfKcxtl0 21733
nnx-0.0.8.dist-info/WHEEL sha256=kLuE8m1WYU0Ig0_YEGrXyTtiJvKPpLpDEiChiNyei5Y 88
nnx-0.0.8.dist-info/RECORD