pypi.org : mytree
Module pytrees that cleanly handle parameter trainability and transformations for JAX models.
Registry
-
Source
- Documentation
- JSON
purl: pkg:pypi/mytree
Keywords:
bijector
, jax
, module
, pytree
, trainability
License: MIT
Latest release: about 2 years ago
First release: about 2 years ago
Downloads: 98 last month
Stars: 3 on GitHub
Forks: 1 on GitHub
See more repository details: repos.ecosyste.ms
Funding links: https://github.com/sponsors/JaxGaussianProcesses
Last synced: about 23 hours ago
Daniel-Dodd
13 packages1,612 downloads