activations
|
Add mobilenet and efficientnet models.
|
2023-05-22 16:39:24 -07:00 |
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
|
Add convnext, inception models
|
2023-05-22 19:13:53 -07:00 |
operations
|
Minor refactor
|
2023-05-23 16:01:22 -07:00 |
random
|
Enable training tests and fix a range of bugs
|
2023-05-19 11:40:17 -07: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 |
trainers
|
Add default optimizer to match tf.keras.
|
2023-05-24 11:15:15 -07:00 |
utils
|
Remove feature space for now
|
2023-05-18 16:07:47 -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 |