Load HuggingFace SafeTensors models to JAX/Flax
Project description
waxjax
Lightweight helper to load HuggingFace SafeTensors checkpoints into JAX/Flax.
Install (latest from source):
pip install git+https://github.com/ik12-b/waxjax.git
Build and publish (recommended to test to TestPyPI first): see RELEASE.md or run locally:
python -m pip install --upgrade build twine
python -m build
twine upload --repository testpypi dist/*
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
waxjax-0.1.0.tar.gz
(8.0 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file waxjax-0.1.0.tar.gz.
File metadata
- Download URL: waxjax-0.1.0.tar.gz
- Upload date:
- Size: 8.0 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.13.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
b3950aa2b21ba02d65d667177df6b6cfe37d24ebf00fa63887d7f613c0b78dd9
|
|
| MD5 |
c35e2ef68a28372358a5b6e3715e0439
|
|
| BLAKE2b-256 |
a9f1b428a5db83266f785ee0d5fc08876a780a6ffc2a336d769428bb4cc62e40
|
File details
Details for the file waxjax-0.1.0-py3-none-any.whl.
File metadata
- Download URL: waxjax-0.1.0-py3-none-any.whl
- Upload date:
- Size: 4.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.13.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
aabc15819eea3624e3ed7808d97d92d2a1e36ee5d149a40fec9dbc998468c56f
|
|
| MD5 |
9ab207a1d27d0a742069b3c6ce43b04f
|
|
| BLAKE2b-256 |
1c4adfb2986f6cef502b9c8fded779abcd172082301a95dec2775c5b59cfd4a9
|
