jax-transformer

View on PyPIReverse Dependencies (0)

0.0.2 jax_transformer-0.0.2-py3-none-any.whl

Wheel Details

Project: jax-transformer
Version: 0.0.2
Filename: jax_transformer-0.0.2-py3-none-any.whl
Download: [link]
Size: 9552
MD5: 3e0eb63b203db5bda9b5e83bf45c0420
SHA256: 265b2ba3c75aab4c877bbd20368324f9bb449c7900f8f1c02350ce209a11bf83
Uploaded: 2024-09-08 01:03:06 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jax-transformer
Version: 0.0.2
Summary: Jax Transformer - Jax
Author: Kye Gomez
Author-Email: kye[at]apac.ai
Home-Page: https://github.com/kyegomez/JaxTransformer
Project-Url: Documentation, https://github.com/kyegomez/JaxTransformer
Project-Url: Repository, https://github.com/kyegomez/JaxTransformer
License: MIT
Keywords: artificial intelligence,deep learning,optimizers,Prompt Engineering
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.9
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10,<4.0
Requires-Dist: flax
Requires-Dist: jax
Description-Content-Type: text/markdown
[Description omitted; length: 5880 characters]

WHEEL

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

RECORD

Path Digest Size
jax_transformer/__init__.py sha256=z1LAq1GW-DNosj3vHQg9eLsHX1pW_wJU8Pf6ybihsDw 379
jax_transformer/embed_layer.py sha256=NUfQG7WHBoXaVZv6fMUw8sRyR0Kgs_JZoBPyfYZQGW0 1435
jax_transformer/lora.py sha256=mCt4sTFe6LED_iPPk35Mt8SbkYNfVrjND6e_zVuppb8 2466
jax_transformer/main.py sha256=-DPL9uQESTB8tt6PEorgtUnc9W7JqQeG6_53B_kmiIc 8118
jax_transformer/output_head.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_transformer/proj.py sha256=_2rdyhPwlOZa56eZPVHW4Pjd6KotUzS45YY_BFuX3lg 877
jax_transformer-0.0.2.dist-info/LICENSE sha256=GHYD0Vn7cqu6O1FibXB1ix_cx8js09uZI1rKjCjW5sY 1074
jax_transformer-0.0.2.dist-info/METADATA sha256=wY-3I3JVFUn437CggCcmH-5OZiKUCxZ-9EdEbEWAwuc 6867
jax_transformer-0.0.2.dist-info/WHEEL sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg 88
jax_transformer-0.0.2.dist-info/RECORD