"Module pytrees" that cleanly handle parameter trainability and transformations for JAX models.
-
Updated
May 11, 2023 - Python
"Module pytrees" that cleanly handle parameter trainability and transformations for JAX models.
Add a description, image, and links to the bijector topic page so that developers can more easily learn about it.
To associate your repository with the bijector topic, visit your repo's landing page and select "manage topics."