|
schedules
|
Initial draft of the codebase.
|
2023-04-09 12:21:45 -07:00 |
|
__init__.py
|
Initial draft of the codebase.
|
2023-04-09 12:21:45 -07:00 |
|
optimizer.py
|
Add fully operational JAX trainer.
|
2023-04-18 14:49:38 -07:00 |
|
sgd.py
|
Add some numpy ops (#1)
|
2023-04-12 11:31:58 -07:00 |
