activations
|
Add Functional model draft.
|
2023-04-12 14:27:30 -07:00 |
backend
|
Add numpy module in jax/ and tensorflow/ (#13)
|
2023-04-18 18:45:30 -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
|
Add numpy module in jax/ and tensorflow/ (#13)
|
2023-04-18 18:45:30 -07:00 |
losses
|
Misc fixes
|
2023-04-17 14:55:17 -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
|
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 fully operational JAX trainer.
|
2023-04-18 14:49:38 -07:00 |
utils
|
Add code stats util
|
2023-04-16 14:54:32 -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 |