mixed-precision-for-JAX
A toolbox for mixed precision training via JAX.
Installation
In a virtualenv (see these instructions if you need to create one):
pip3 install mixed-precision-for-jax
Releases
| Version | Released | Bullseye Python 3.9 |
Bookworm Python 3.11 |
Files |
|---|---|---|---|---|
| 0.1.5 | 2025-05-26 | |||
|
|
||||
| 0.1.4 | 2025-05-26 | |||
|
|
||||
| 0.1.3 | 2025-05-26 | |||
|
|
||||
| 0.1.2 | 2025-05-20 | |||
|
|
||||
| 0.1.1 | 2025-05-20 | |||
|
|
||||
| 0.1.0 | 2025-05-20 | |||
|
|
||||
Issues with this package?
- Search issues for this package
- Package or version missing? Open a new issue
- Something else? Open a new issue