List Question
4 TechQA 2024-09-18 13:36:08What is the main difference between flax (google) and dm-haiku (deepmind)?
7.8k views
Asked by discort
AttributeError: module 'jax.tree_util' has no attribute 'PyTreeDef' when import haiku
583 views
Asked by Manyerror
Mixed Precision Training using Jax
635 views
Asked by Jankins21
Migration from haiku: Alternative to Haiku's PRNGSequence?
78 views
Asked by Hylke