Francois Chollet
1983d1a402
Add JAX evaluate flow.
2023-04-19 18:37:25 -07:00
Francois Chollet
aa4078899d
Add TF evaluate() flow.
2023-04-19 16:25:56 -07:00
Francois Chollet
ee5be68ce9
Fix progbar glitch
2023-04-19 15:24:35 -07:00
Francois Chollet
4bf04920f2
Fix jax jit_compile setting
2023-04-19 14:35:28 -07:00
Francois Chollet
2fa38c1071
Fix JAX trainer correctness issue and add test
2023-04-19 13:50:22 -07:00
Francois Chollet
efe400a388
Rename data_adapter_utils
2023-04-18 21:11:21 -07:00
Chen Qian
f6df67f2d2
Add numpy module in jax/ and tensorflow/ ( #13 )
...
* Add jax/numpy and tensorflow/numpy
* refactor code
* more
* even better
2023-04-18 18:45:30 -07:00
Francois Chollet
dd144b6097
Fix jax dropout and start tracking all state in layer.variables.
2023-04-18 16:21:27 -07:00
Francois Chollet
89b2ea5977
Add fully operational JAX trainer.
2023-04-18 14:49:38 -07:00
Francois Chollet
0f97daf7a7
Jax trainer checkpoint.
2023-04-18 13:02:29 -07:00
Francois Chollet
d0f3799dd8
Move stateless scope to its own file.
2023-04-18 08:52:21 -07:00
Francois Chollet
dc0f3467fc
Add draft of JAX trainer
2023-04-17 21:26:04 -07:00
Francois Chollet
df6990f41a
Misc fixes + add support for JAX array input
2023-04-17 17:47:22 -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
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
