|
activations
|
Add Functional model draft.
|
2023-04-12 14:27:30 -07:00 |
|
backend
|
Add draft of JAX trainer
|
2023-04-17 21:26:04 -07:00 |
|
callbacks
|
Make progbar logger work
|
2023-04-17 16:11:30 -07:00 |
|
constraints
|
Initial draft of the codebase.
|
2023-04-09 12:21:45 -07:00 |
|
initializers
|
Add draft of JAX trainer
|
2023-04-17 21:26:04 -07:00 |
|
layers
|
Misc fixes
|
2023-04-17 14:55:17 -07:00 |
|
losses
|
Misc fixes
|
2023-04-17 14:55:17 -07:00 |
|
metrics
|
Add draft of JAX trainer
|
2023-04-17 21:26:04 -07:00 |
|
models
|
Add draft of JAX trainer
|
2023-04-17 21:26:04 -07:00 |
|
operations
|
Misc fixes.
|
2023-04-17 15:41:48 -07:00 |
|
optimizers
|
Add draft of JAX trainer
|
2023-04-17 21:26:04 -07:00 |
|
regularizers
|
Fix some tests
|
2023-04-12 20:40:23 -07:00 |
|
saving
|
Initial draft of the codebase.
|
2023-04-09 12:21:45 -07:00 |
|
testing
|
Add some numpy ops (#1)
|
2023-04-12 11:31:58 -07:00 |
|
trainers
|
Add draft of JAX trainer
|
2023-04-17 21:26:04 -07:00 |
|
utils
|
Add code stats util
|
2023-04-16 14:54:32 -07:00 |
|
__init__.py
|
Add some numpy ops (#1)
|
2023-04-12 11:31:58 -07:00 |
|
api_export.py
|
Add some numpy ops (#1)
|
2023-04-12 11:31:58 -07:00 |
