Package search
jmp
- Summary: JMP is a Mixed Precision library for JAX.
- Author: DeepMind
- Homepage: https://github.com/deepmind/jmp
- Source: https://github.com/deepmind/jmp
- Repo profile
- Number of releases: 4
- First release: 0.0.1 on 2019-09-08T20:49:45
- Latest release: 0.0.4 on 2023-01-30T12:47:11
| Dependency | Optional |
|---|---|
| dataclasses | false |
| numpy | false |
| jax | true |
| jaxlib | true |