callbacks
|
Fix nits
|
2023-05-15 12:37:52 -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 |
losses
|
Fix numerical test
|
2023-05-15 16:33:35 -07:00 |
metrics
|
Add JAX RNN support.
|
2023-05-16 10:52:14 -07:00 |
mixed_precision
|
Refactor backend.
|
2023-05-03 15:33:40 -07:00 |
models
|
Rename pytorch to torch.
|
2023-05-15 13:56:52 -07:00 |
operations
|
Export operations (#176)
|
2023-05-16 20:35:11 -07:00 |
random
|
Adds Random Unit Tests (#174)
|
2023-05-16 00:21:36 -05:00 |
regularizers
|
Add normalize() to numerical utils.
|
2023-04-29 15:08:38 -07:00 |
saving
|
Add Lambda layer.
|
2023-05-14 22:45:20 -07:00 |
testing
|
Add layers.MultiHeadAttention (#169)
|
2023-05-16 18:22:42 -07:00 |
utils
|
Add Conv LSTM layers
|
2023-05-17 16:06:01 -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 |