5 Commits

Author	SHA1	Message	Date
Francois Chollet	fa2ea78d3f	Fix typos	2023-05-02 22:44:46 -07:00
Francois Chollet	4bf04920f2	Fix jax jit_compile setting	2023-04-19 14:35:28 -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	1a45e5cd17	Add demos	2023-04-18 15:46:57 -07:00