jaxformers

View on PyPIReverse Dependencies (0)

0.0.1.dev1 jaxformers-0.0.1.dev1-py3-none-any.whl

Wheel Details

Project: jaxformers
Version: 0.0.1.dev1
Filename: jaxformers-0.0.1.dev1-py3-none-any.whl
Download: [link]
Size: 8185
MD5: bd826faeede1505d54722655705361df
SHA256: 5434cf8de9dd44f74c6e7318909753420458e474e5f767374038239f0fa47bcb
Uploaded: 2023-02-03 12:20:16 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jaxformers
Version: 0.0.1.dev1
Summary: 'Attention is all you need' in JAX (Flax)
Author-Email: Alvaro Bartolome <alvarobartt[at]yahoo.com>
Project-Url: Documentation, https://alvarobartt.github.io/jaxformers
Project-Url: Issues, https://github.com/alvarobartt/jaxformers/issues
Project-Url: Source, https://github.com/alvarobartt/jaxformers
Classifier: Development Status :: 4 - Beta
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: Implementation :: CPython
Classifier: Programming Language :: Python :: Implementation :: PyPy
Requires-Python: >=3.7
Requires-Dist: chex (~=0.1.5)
Requires-Dist: flax (~=0.6.3)
Requires-Dist: jaxlib (~=0.3.25)
Requires-Dist: jax (~=0.3.25)
Requires-Dist: mkdocs-git-revision-date-localized-plugin (~=1.1.0); extra == "docs"
Requires-Dist: mkdocs-material (~=8.5.4); extra == "docs"
Requires-Dist: mkdocstrings[python] (~=0.19.0); extra == "docs"
Requires-Dist: mkdocs (~=1.4.0); extra == "docs"
Requires-Dist: black (~=22.10.0); extra == "quality"
Requires-Dist: pre-commit (~=2.20.0); extra == "quality"
Requires-Dist: ruff (~=0.0.194); extra == "quality"
Requires-Dist: pytest (~=7.1.2); extra == "tests"
Provides-Extra: docs
Provides-Extra: quality
Provides-Extra: tests
Description-Content-Type: text/markdown
License-Expression: MIT
License-File: LICENSE
[Description omitted; length: 72 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.12.2
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
jaxformers/__init__.py sha256=E3vSDG2vzhz7TwQ2Ev0HmrRHtlobDAiv4Z_VCahj5FM 147
jaxformers/attention.py sha256=5YUEMGfUNXPoB0lg4aj9GAzM1h69t45XdEL_55rCQEI 4397
jaxformers/blocks.py sha256=O0OMQkrULqdYhmznZu8FCN-YWU6zP_uyNVvaODS0wxQ 3997
jaxformers/decoder.py sha256=Kdx6BlWei6iNYi0UnIP360nuIeHRHcvdZMfvbpxwFvg 1741
jaxformers/encoder.py sha256=diKj4G6SMpBfmi8Nbnk76shloV2qX_Ev6WpcsYnBo0U 926
jaxformers/encoding.py sha256=o9hN6AQEZ4E63cUSVKR5Y6tmhP6q5UG9u1CkhJCqDu4 1357
jaxformers/normalization.py sha256=a76l_dG5jHnWjXXjFuqZLU4noAfkBX3Ux5SI6-9h5C8 601
jaxformers/py.typed sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jaxformers/transformer.py sha256=2ztLw6_U23htftdSU71CZgxZTSdp6T0xs-aSx3nRc7Q 2673
jaxformers/utils.py sha256=7drTqwCx3ATQX-2Tb3TxKN5FancqVYfFPQwgbGsM3QA 169
jaxformers-0.0.1.dev1.dist-info/METADATA sha256=XZBAwpwEVSfLdk9ObWB4ij9JaONAqUdGTCbmKhbJqcQ 1611
jaxformers-0.0.1.dev1.dist-info/WHEEL sha256=hKi7AIIx6qfnsRbr087vpeJnrVUuDokDHZacPPMW7-Y 87
jaxformers-0.0.1.dev1.dist-info/licenses/LICENSE sha256=pHCbcfMj4Zai8CwBPbjpRztb93igu9pZdCj_CdYYGr8 1104
jaxformers-0.0.1.dev1.dist-info/RECORD