activations
|
Add Functional model draft.
|
2023-04-12 14:27:30 -07:00 |
backend
|
Add JAX evaluate flow.
|
2023-04-19 18:37:25 -07:00 |
callbacks
|
Make progbar logger work
|
2023-04-17 16:11:30 -07:00 |
constraints
|
Export initializers
|
2023-04-19 12:52:58 -07:00 |
initializers
|
Export initializers
|
2023-04-19 12:52:58 -07:00 |
layers
|
Fix JAX trainer correctness issue and add test
|
2023-04-19 13:50:22 -07:00 |
losses
|
Fix indent
|
2023-04-18 21:18:55 -07:00 |
metrics
|
Add fully operational JAX trainer.
|
2023-04-18 14:49:38 -07:00 |
models
|
Add fully operational JAX trainer.
|
2023-04-18 14:49:38 -07:00 |
optimizers
|
Add fully operational JAX trainer.
|
2023-04-18 14:49:38 -07:00 |
regularizers
|
Export initializers
|
2023-04-19 12:52:58 -07:00 |
saving
|
Initial draft of the codebase.
|
2023-04-09 12:21:45 -07:00 |
trainers
|
Add JAX evaluate flow.
|
2023-04-19 18:37:25 -07:00 |
utils
|
Fix progbar glitch
|
2023-04-19 15:24:35 -07:00 |
__init__.py
|
Add demos
|
2023-04-18 15:46:57 -07:00 |
api_export.py
|
Add some numpy ops (#1)
|
2023-04-12 11:31:58 -07:00 |