Wheelodex » Projects » flax-extra

flax-extra

View on PyPI — Reverse Dependencies (0)

0.0.2 flax_extra-0.0.2-py3-none-any.whl

Wheel Details

Project: flax-extra
Version: 0.0.2
Filename: flax_extra-0.0.2-py3-none-any.whl
Download: [link]
Size: 61626
MD5: 9364a7fe1a0a05bc4d19c0b3c26279be
SHA256: aa793d3364024cf38ba158212e6ef96ae4a76633030552ca463a0030b463ef19
Uploaded: 2021-10-07 22:33:12 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: flax-extra
Version: 0.0.2
Summary: The package provides extra flexibility to Flax using ideas originated at Trax
Author: Andrei Nesterov
Author-Email: ae.nesterov[at]gmail.com
Home-Page: https://github.com/manifest/flax-extra
Project-Url: Documentation, https://flax-extra.readthedocs.io
Project-Url: Source Code, https://github.com/manifest/flax-extra
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3.9
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Typing :: Typed
Requires-Python: >=3.9
Requires-Dist: redex (>=0.1)
Requires-Dist: flax (>=0.3)
Requires-Dist: einops (>=0.3)
Requires-Dist: tensorflow (>=2.6)
Requires-Dist: numpy; extra == "development"
Requires-Dist: pylint (==2.10); extra == "development"
Requires-Dist: sphinx; extra == "docs"
Requires-Dist: furo; extra == "docs"
Requires-Dist: nbsphinx; extra == "docs"
Requires-Dist: ipykernel; extra == "docs"
Provides-Extra: development
Provides-Extra: docs
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 1976 characters]

WHEEL

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

RECORD

Path Digest Size
flax_extra/__init__.py sha256=R53PG61WeA5QRgOhLPcZgZIsXLVQNqsj-oGReU7D0gY 361
flax_extra/batch.py sha256=jwRO0v2NSpzHBpNh02julzvELVjLUhEM9bV1PiLFBHI 1957
flax_extra/console.py sha256=H3YLRzU0JkehEwEwMFmec29xgc-nFbXp3sqfV9rJ0s8 600
flax_extra/py.typed sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE 58
flax_extra/random.py sha256=9iE_HetVli8jzRQbG-m1jaI-mZ5nbR4fLIXY79nKylk 2078
flax_extra/util.py sha256=u_luWmLqRHku41tOtAjX5ZsrTs883INr5LMskRM48HE 1268
flax_extra/version.py sha256=qYq19k3S5-8F8MG3D8QOmJIWydzKll7VPSwzs8H3OWs 51
flax_extra/checkpoint/__init__.py sha256=7U1ClJ3U_3rXfn_TXr8mFo_qjMNpzVIN2xs-32KpB4Y 1113
flax_extra/checkpoint/_best_checkpoint_file_reader.py sha256=3pooVCAPAHhKezjtuH89VW6BcXcqwa7WMs9dPmIh5dU 2237
flax_extra/checkpoint/_best_checkpoint_file_writer.py sha256=lBJeeC4L7pt6Kv4ytkWJNlNYJpjmG7MebsP3hQAA3Pw 6795
flax_extra/checkpoint/_checkpoint.py sha256=0ZMYJ5Dtfn00K7M09nhGKIFP-VPfCdpyh9hgaZTt9G8 1002
flax_extra/checkpoint/_checkpoint_file.py sha256=So3kT5xP8W5udPiML-dVh4riAlYvTtWxgEBl6v76Q8o 1283
flax_extra/checkpoint/_checkpoint_file_reader.py sha256=CfP1jZtKQiwZVal9RifI_6Wg8RBlY0zuVhtpOjhFUfg 2698
flax_extra/checkpoint/_checkpoint_file_writer.py sha256=ft-pG5214XuTAo1UgE_I1QV5uJfmr-QXF5UqJaVkQF8 2821
flax_extra/checkpoint/_summary.py sha256=qmWE08KUAD0Njf4P7xed2ElURcqf4KUsQAUyuDcHqNk 395
flax_extra/checkpoint/_summary_logger.py sha256=iEMyLSkFPD8G2j6qGaK2pEddYbXrposOd2lQcHL6mQs 1678
flax_extra/checkpoint/_summary_writer.py sha256=pl6YqPcnA11GRMH6VJQbCLGb_e3uAbx_czllDcCu4eA 873
flax_extra/combinator/__init__.py sha256=cmdx1SobNVIViKtXLihcez70KJNiLfCSq9ghF_FH0zI 519
flax_extra/combinator/_concatenate.py sha256=J68Od1OK-ypMYfxArZrlQ1I9VRVFaS0FY_Gl-Hrb9dQ 1322
flax_extra/data/__init__.py sha256=NThypTSjyIYwUGDc6PtAnvQgazAFo2R9m_CRKKvty9w 184
flax_extra/data/_bytes_tokenizer.py sha256=D8lbWooKt9F5ChbdfIKMAvboBJlKc26RZwO6Th8VjHE 2872
flax_extra/evaluation/__init__.py sha256=nfj5Pb-XT2TtWwFAcQvTFAG_FaeIV9CackhITjB8Itg 130
flax_extra/evaluation/_loop.py sha256=QToWNQm2FyGmf6i5Xqebia_W1sTIz6HOFhBQ83QVbbA 7720
flax_extra/layer/__init__.py sha256=l_suyw0vNBMekIcEJckX6VV_0t1QhInhkM0AfVa9_-o 2026
flax_extra/layer/_attention.py sha256=A-_xnhd_Q45Wley-Dy2RoobP0xgTdsSTbhXP_2GBSxw 16876
flax_extra/layer/_decoding.py sha256=vmEndQP1gMt1ccIzzGTT06gFjBGWkCjaPpVWAvbif_w 2049
flax_extra/layer/_embedding_decoding.py sha256=Bkrb6qxF7F8n2CmDaYaPgeZv99F8m0de-b9Q5SVLYXM 1474
flax_extra/layer/_encoding.py sha256=_3EFqPuWZr9dkKKpCNwt1Jcl8vO7P90Za3lKWRK9b4s 3553
flax_extra/layer/_feedforward.py sha256=Dl8FF-BdrEx8e2U4jfSAcBB89wZjYL5ckRxbi3SikJM 2082
flax_extra/layer/_fourier_positional_encoding.py sha256=lERaQV0URN5doe_0GroaeOMSC8wTjMRp_DaBqSLC1cw 8207
flax_extra/layer/_lstm.py sha256=I5TIiy6oyDpU-VcCqaAPlMNKg33NC6k01WX7BDqWRe0 1592
flax_extra/layer/_multimodal_decoding.py sha256=5XT1bODapu_50SLDydoBV5Zhq2lwK70ssEaE97Xoh10 3651
flax_extra/layer/_multimodal_encoding.py sha256=RQlKcari8KcK5VvfjMSNnDB59CKSawMeeb3-_oa8BHY 6252
flax_extra/layer/_multimodal_positional_encoding.py sha256=yF56Cc7q77zButg2_RYyPbzE-E-6MU7Pa7y6pJ1djac 5572
flax_extra/layer/_trainable_positional_encoding.py sha256=thsCdYcYusjsTg7jgFmnmqlerwSukadrS7YtaGPihhM 2617
flax_extra/layer/_trainable_positional_masking.py sha256=Mao-144E7k4EwvF5VcLlcJlPNvGc-0oZh5ktPxNBG6I 1569
flax_extra/layer/_trainable_positional_padding.py sha256=6HHKtfgA35bM27kyIrWKOylJDduUGd-9lBBgGG1AlYY 1774
flax_extra/layer/io.py sha256=ex1SqRgoMFkQUp_t6EfNzOeo3I2qZfr8eoYW8CVLYAM 2033
flax_extra/model/__init__.py sha256=kE4_S_9YBFCWRIUcXRIJvuLUc0ckG2oEIlS6uy7tr_Q 166
flax_extra/model/perceiver/__init__.py sha256=nZMgNcIBt_gc1ku4TUXHCNuoCdI0aNGHsEjL1G3L6zE 477
flax_extra/model/perceiver/_cross_attention_block.py sha256=I6zohp6XhF0qYh_gGmxwXGXWbYNc5Owa0LQisqE7YqI 3463
flax_extra/model/perceiver/_decoder.py sha256=3xmROYUudmUNts1juS5kCmW-6GFh-1tMUWkxffAs6qc 2925
flax_extra/model/perceiver/_encoder.py sha256=-Ix14kvXfVE5AW9TPgg7GIWsuK-gdmGbPKtVdXBTBNo 2982
flax_extra/model/perceiver/_io.py sha256=NRO2aPP9QY2FDHO3dwsZSGAgYVuR0AD_ERXmkGULVHE 9350
flax_extra/model/perceiver/_processor.py sha256=eb_9tDwdKlF1y_Waa9O1PoZ8lds51_8sC2f1kzVmwMg 2569
flax_extra/model/perceiver/_self_attention_block.py sha256=TfI4mg8PONKUPeaOpaFQH97Ah23dQS6_H-14_nPgvlQ 2608
flax_extra/model/rnn/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
flax_extra/model/rnn/_lm.py sha256=RrdvgR4-EtF9emloxIrX5EaepC_UjwP2dyfrQ5306ac 1376
flax_extra/operator/__init__.py sha256=BQg7FfY-81uFQrPPF7wJcLPiBCSCrWWXVYMWGJU8-K8 270
flax_extra/operator/_rearrange.py sha256=ru1NGnpLDkHQvTUKsQh0MkhFDE6zCfcCEKk1Y0C6MQw 868
flax_extra/operator/_reshape.py sha256=m9L7gequtnxlQ0d7QlK0hD7vQ1ELIej066nF4QdBYCc 716
flax_extra/operator/_shift_right.py sha256=JU_j3ZTPDO1WmS5kHDEmvaJ0Dwv-Fbw0ejwhwdmrSdA 1214
flax_extra/training/__init__.py sha256=oGr3bwRrYYYGR9G3LlEerC0Y-Iu21uoBixd4NyZUxsg 130
flax_extra/training/_loop.py sha256=Wol4ipqHuA5763g1V9aZlkT9ExEvEWj7hMeTJ05yoGM 19312
flax_extra-0.0.2.dist-info/LICENSE sha256=ap9vUizhaJw4Z3pjaSHoZabb04RDq2wNYPC5wifabUk 11345
flax_extra-0.0.2.dist-info/METADATA sha256=StQW1d5HnRVwSkOnCGekWUFO_eSeLFPl3H9BDScqtzE 3307
flax_extra-0.0.2.dist-info/WHEEL sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o 92
flax_extra-0.0.2.dist-info/top_level.txt sha256=vBkctqtmseD6uWb4mnipH1ghVVsUKHXBtYh3Y0gBMb4 11
flax_extra-0.0.2.dist-info/RECORD — —

top_level.txt

flax_extra