Srimukh Sripada 818cda16bc Support LR schedulers (#334) ...

* Add a few LR schedulers

* Move parents's constructor call to the top

* Fix docstring

* refactor optimizers into two files

* add docs

* nit

* Fix Callable type annotation for python 3.8

---------

Co-authored-by: Awni Hannun <awni@apple.com>
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>

2024-02-15 11:26:20 -08:00

..

nn

Pooling layers (#357)

2024-02-12 22:08:13 -08:00

optimizers

Support LR schedulers (#334)

2024-02-15 11:26:20 -08:00

array.rst

Adds round op and primitive (#203)

2023-12-18 11:32:48 -08:00

data_types.rst

Fix data_types.rst uint64 (#406)

2024-01-09 06:40:10 -08:00

devices_and_streams.rst

Adds device context manager (#679)

2024-02-14 14:14:58 -08:00

fft.rst

awni's commit files

2023-11-29 10:30:41 -08:00

linalg.rst

QR factorization (#310)

2024-01-26 09:27:31 -08:00

nn.rst

Common neural network initializers nn.initializers (#456)

2024-01-23 06:47:20 -08:00

ops.rst

Implement diagonal operator (#562)

2024-01-30 09:45:48 -08:00

optimizers.rst

Support LR schedulers (#334)

2024-02-15 11:26:20 -08:00

random.rst

Sort some APIs docs by names (a-z) (#472)

2024-01-16 19:37:50 -08:00

transforms.rst

Compile docs (#653)

2024-02-08 11:21:50 -08:00

tree_utils.rst

jagrit's commit files

2023-11-29 10:52:08 -08:00