Awni Hannun e5851e52b1 Add move and swap axis, and vmap for slice, concat, and gather (#158) ...

* add move and swap axis, and vmap for slice, concat, and gather

2023-12-14 12:59:12 -08:00

..

nn

awni's commit files

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

array.rst

angelos's commit files

2023-11-29 10:42:59 -08:00

data_types.rst

jagrit's commit files

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

devices_and_streams.rst

jagrit's commit files

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

fft.rst

awni's commit files

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

nn.rst

Activations LeakyReLU / PReLU / Softplus / Mish (#109)

2023-12-11 19:40:57 -08:00

ops.rst

Add move and swap axis, and vmap for slice, concat, and gather (#158)

2023-12-14 12:59:12 -08:00

optimizers.rst

awni's commit files

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

random.rst

awni's commit files

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

transforms.rst

jagrit's commit files

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

tree_utils.rst

jagrit's commit files

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