Awni Hannun aac2f9fb61 Improve profiling with gpu tracing (#969) ...

* improve profiling with gpu tracing

* fix for linux

* nit

* doc fix

* fix example

2024-04-07 21:47:43 -07:00

..

nn

add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)

2024-03-25 12:32:59 -07:00

optimizers

Add linear warmup and schedule joining for use with existing schedules (#721)

2024-02-26 07:28:48 -08:00

array.rst

add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)

2024-03-25 12:32:59 -07:00

data_types.rst

add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)

2024-03-25 12:32:59 -07:00

devices_and_streams.rst

Adds device context manager (#679)

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

fast.rst

post nanobind docs fixes and some updates (#889)

2024-03-24 15:03:27 -07: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

metal.rst

Improve profiling with gpu tracing (#969)

2024-04-07 21:47:43 -07:00

nn.rst

Common neural network initializers nn.initializers (#456)

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

ops.rst

add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)

2024-03-25 12:32:59 -07: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