backend
|
Fix jax bug + annoying TF warning
|
2023-05-13 22:54:25 -07:00 |
constraints
|
Add bulk of serialization logic.
|
2023-04-21 10:00:32 -07:00 |
datasets
|
Touch ups
|
2023-05-13 17:33:48 -07:00 |
initializers
|
Increase initializer test coverage
|
2023-05-09 16:41:25 -07:00 |
layers
|
Fix conv layer signatures
|
2023-05-14 19:11:16 -07:00 |
losses
|
Populate metrics and losses for deserialization
|
2023-05-09 16:26:12 -07:00 |
mixed_precision
|
Refactor backend.
|
2023-05-03 15:33:40 -07:00 |
models
|
Add support for h5 weights loading.
|
2023-05-08 11:34:55 -07:00 |
operations
|
Add package build system.
|
2023-05-13 20:17:03 -07:00 |
regularizers
|
Add normalize() to numerical utils.
|
2023-04-29 15:08:38 -07:00 |
testing
|
Add pad_sequences util.
|
2023-05-14 10:45:27 -07:00 |
trainers
|
fix lint (#168)
|
2023-05-14 18:41:50 +00:00 |
utils
|
Add pad_sequences util.
|
2023-05-14 10:45:27 -07:00 |
__init__.py
|
Touch ups
|
2023-05-13 17:33:48 -07:00 |
api_export.py
|
Added initial Adam tests
|
2023-04-24 22:07:56 +00:00 |