Piotr Rybiec 6a665ea6ed Dilation for convolutional layers (#766) ...

* add dilation parameter to Conv1d layer

* space here too

* add conv1d dilation test

* add dilation parameter for Conv2d layer

* conv2d dilation test

2024-03-04 06:43:00 -08:00

..

layers

Dilation for convolutional layers (#766)

2024-03-04 06:43:00 -08:00

__init__.py

Common neural network initializers nn.initializers (#456)

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

init.py

updated calls to use loc &scale (#643)

2024-02-08 09:01:59 -08:00

losses.py

Feat: Add weights argument in BCE Loss and tests (#620)

2024-02-07 09:39:52 -08:00

utils.py

Custom VJP and checkpointing (#541)

2024-01-30 16:04:45 -08:00