Name | Size | Last Modified |
|---|---|---|
| README.md | 9 KB | |
| batchnorm_jax.ipynb | 97 KB | |
| batchnorm_torch.ipynb | 127.2 KB | |
| cifar10_cnn_lightning.ipynb | 24.2 MB | |
| cnn_cifar_pytorch.ipynb | 170 KB | |
| cnn_mnist_tf.ipynb | 114.5 KB | |
| conv2d_jax.ipynb | 17.2 KB | |
| conv2d_torch.ipynb | 16.7 KB | |
| densenet_jax.ipynb | 62.4 KB | |
| densenet_torch.ipynb | 97.1 KB | |
| layer_norm_jax.ipynb | 3.5 KB | |
| layer_norm_torch.ipynb | 4.3 KB | |
| lenet_jax.ipynb | 52.7 KB | |
| lenet_torch.ipynb | 55.6 KB | |
| resnet_jax.ipynb | 57 KB | |
| resnet_torch.ipynb | 88.9 KB | |
| supplementary/ | - | |
| transposed_conv_jax.ipynb | 3.1 KB | |
| transposed_conv_torch.ipynb | 3.6 KB |
