Ramesh Sampath
42bdabf76a
Resolve shape via ops in broadcast_to operation ( #547 )
2023-07-19 15:04:11 -05:00
Francois Chollet
f0b99c51cf
Fix code style
2023-06-28 16:04:13 -07:00
Francois Chollet
08da564985
Namespace nit: operations -> ops
2023-06-28 15:36:45 -07:00
Francois Chollet
cef6dfae9c
Add jax train step guide
2023-06-28 11:58:53 -07:00
Francois Chollet
deeb4ad9ab
Add JAX custom training loop guide.
2023-06-27 11:57:24 -07:00
Francois Chollet
0cdd15d917
Minor fixes.
2023-06-14 15:36:31 -07:00
Francois Chollet
c2d7c51aba
Extensive refactor of tracking logic.
2023-06-05 18:39:26 -07:00
Francois Chollet
77742304ac
Fix minor issues
2023-05-28 18:28:19 -07:00
Francois Chollet
083c817788
Add missing docstrings
2023-05-26 14:11:03 -07:00
Francois Chollet
93f5c6cd7d
Add Metric docstring
2023-05-26 09:13:08 -07:00
Francois Chollet
3db16c668c
Big rethink of Variable API
2023-04-26 10:29:40 -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
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