Merge branch 'haiku-jax' into 'develop'
Add support for Jax / Haiku See merge request !32
Showing
- declearn/model/__init__.py 4 additions, 0 deletionsdeclearn/model/__init__.py
- declearn/model/haiku/__init__.py 31 additions, 0 deletionsdeclearn/model/haiku/__init__.py
- declearn/model/haiku/_model.py 521 additions, 0 deletionsdeclearn/model/haiku/_model.py
- declearn/model/haiku/_vector.py 170 additions, 0 deletionsdeclearn/model/haiku/_vector.py
- declearn/model/haiku/utils/__init__.py 25 additions, 0 deletionsdeclearn/model/haiku/utils/__init__.py
- declearn/model/haiku/utils/_gpu.py 76 additions, 0 deletionsdeclearn/model/haiku/utils/_gpu.py
- declearn/model/torch/__init__.py 1 addition, 1 deletiondeclearn/model/torch/__init__.py
- declearn/test_utils/_vectors.py 9 additions, 1 deletiondeclearn/test_utils/_vectors.py
- pyproject.toml 8 additions, 2 deletionspyproject.toml
- test/functional/test_regression.py 32 additions, 8 deletionstest/functional/test_regression.py
- test/model/test_haiku.py 295 additions, 0 deletionstest/model/test_haiku.py
- test/model/test_tflow.py 3 additions, 3 deletionstest/model/test_tflow.py
Loading
Please register or sign in to comment