keras/keras_core/backend/jax
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
..
__init__.py Add numpy module in jax/ and tensorflow/ (#13) 2023-04-18 18:45:30 -07:00
numpy.py Add numpy module in jax/ and tensorflow/ (#13) 2023-04-18 18:45:30 -07:00
random.py Add numpy module in jax/ and tensorflow/ (#13) 2023-04-18 18:45:30 -07:00
trainer.py Add fully operational JAX trainer. 2023-04-18 14:49:38 -07:00