Francois Chollet
89b2ea5977
Add fully operational JAX trainer.
2023-04-18 14:49:38 -07:00
Francois Chollet
dc0f3467fc
Add draft of JAX trainer
2023-04-17 21:26:04 -07:00
Chen Qian
eabdb87f9f
Add some numpy ops ( #1 )
...
* Add numpy ops (initial batch) and some config
* Add unit test
* fix call
* Revert "fix call"
This reverts commit 6748ad183029ff4b97317b77ceed8661916bb9a0.
* full unit test coverage
* fix setup.py
2023-04-12 11:31:58 -07:00
François Chollet
4a8d8c6117
Merge pull request #2 from keras-team/add_initializers
...
Add RandomNormal and RandomUniform Initializers
2023-04-12 11:00:14 -07:00
AakashKumarNain
4b018c0560
add tests for RandomNormal and RandomUniform initializers
2023-04-12 23:23:38 +05:30
Francois Chollet
6544d6b850
Reformat with black line_length=80
2023-04-12 10:52:34 -07:00
Francois Chollet
377fcdb91e
Add export decorators for base classes
2023-04-09 12:53:37 -07:00
Francois Chollet
701d4cf7bc
Merge engine namespace into specialized namespaces.
2023-04-09 12:46:00 -07:00
Francois Chollet
160473c474
Initial draft of the codebase.
2023-04-09 12:21:45 -07:00
