Facilitates machine learning development for JAX.
Project description
EasyJax
work-in-progress.
EasyJax is a python package that facilitates machine learning development for JAX. It does that by providing:
- A high-level API for machine learning workflows in JAX (specifically a trainer, experiment parent class).
- Several machine learning specific utilities for working with JAX (e.g.,
ml.update_step).
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
easyjax-0.1.3.tar.gz
(2.7 kB
view details)
Built Distribution
File details
Details for the file easyjax-0.1.3.tar.gz.
File metadata
- Download URL: easyjax-0.1.3.tar.gz
- Upload date:
- Size: 2.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: poetry/1.2.2 CPython/3.8.5 Darwin/20.1.0
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 66ca412ac48212b5426d6a8a40dff1029fd71589dd4dd054179aff6628b64da9 |
|
| MD5 | dff80a457f08087720212a6457bc5f9a |
|
| BLAKE2b-256 | a16ae48333b02aab53c880b5f7d4c3b5fed62e1f774e69b9312d4890152e6b27 |
File details
Details for the file easyjax-0.1.3-py3-none-any.whl.
File metadata
- Download URL: easyjax-0.1.3-py3-none-any.whl
- Upload date:
- Size: 3.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: poetry/1.2.2 CPython/3.8.5 Darwin/20.1.0
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 4fe3c0744bca97c09224793b14f0edf7320c4bafaa452125914adb685ae01449 |
|
| MD5 | bde4ca21adfec4de6769352ffee4c637 |
|
| BLAKE2b-256 | 3f894bcebead200e43beca88277c27258e526eaa224ecf92caa6102ffccafcfa |
