jax-lorax

View on PyPIReverse Dependencies (0)

0.3.1 jax_lorax-0.3.1-py3-none-any.whl

Wheel Details

Project: jax-lorax
Version: 0.3.1
Filename: jax_lorax-0.3.1-py3-none-any.whl
Download: [link]
Size: 7621
MD5: 531fd3e49e6f0940ff30ea410462f6a3
SHA256: a55446c69fb12909ceda8f835f992d6e4ece3de864b698f507f5de2be97ae4f0
Uploaded: 2024-02-26 03:01:30 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jax-lorax
Version: 0.3.1
Summary: A JAX transform which applies LoRA to arbitrary JAX functions/models
Author: Davis Yoshida
Author-Email: dyoshida[at]ttic.edu
License: MIT
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
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 :: 3.12
Requires-Python: >=3.8,<4.0
Requires-Dist: jax (<0.5.0,>=0.4.6)
Requires-Dist: jaxlib (<0.5.0,>=0.4.6)
Requires-Dist: qax (>=0.3.1)
Description-Content-Type: text/markdown
[Description omitted; length: 4465 characters]

WHEEL

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

RECORD

Path Digest Size
lorax/__init__.py sha256=8dFW_rRKzhII9JPQsL3YFNkBQOuPgQyosPkmkwcBlFw 179
lorax/constants.py sha256=41NG1LCTZ27ec8s8H11_9EYs1NrA4m_6JwD-KMsIgrs 31
lorax/helpers.py sha256=DOnxCVExMyNCyDJtCfQHBVlilHN9ILvx4RFzATqTqh4 4365
lorax/transform.py sha256=txEx343MWjv-GaKs7z-NmrRkvxZNDVyXqeHNqfv83rM 5400
jax_lorax-0.3.1.dist-info/LICENSE sha256=pMmSYruATCUahaoDOjhqlM5YTROiajitVUX_HS84jE0 1069
jax_lorax-0.3.1.dist-info/METADATA sha256=HjsEiNW2gKp9-w5sWh5QRwD_mqzi1VmR97XcGcrzZYc 5188
jax_lorax-0.3.1.dist-info/WHEEL sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg 88
jax_lorax-0.3.1.dist-info/RECORD