Metadata-Version: 2.1
Name: jaxformers
Version: 0.0.1.dev0
Summary: 'Attention is all you need' in JAX (Flax)
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
Author-email: Alvaro Bartolome <alvarobartt@yahoo.com>
License-Expression: MIT
License-File: LICENSE
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
Provides-Extra: docs
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'
Provides-Extra: quality
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'
Provides-Extra: tests
Requires-Dist: pytest~=7.1.2; extra == 'tests'
Description-Content-Type: text/markdown

# 🧙🏻‍♂️ `jaxformers`

Reference: https://e2eml.school/transformers.html
