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=MeZ0MU2IhSmgAE-rfFRQeRrSUSArJjgk9e1bFvN8f3w,22437
nnx-0.0.7.dist-info/METADATA,sha256=YZKCUEyQ0fVxUpxh8s0JAki-Uw5ovLK8wiBnMaUcCTU,21733
nnx-0.0.7.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
nnx-0.0.7.dist-info/LICENSE,sha256=nEeqZTgRHhGmbWc6BWQ5ZWubB-K7KxcGA4mU4Vwc98c,1072
nnx-0.0.7.dist-info/RECORD,,
