attention
|
Enable training tests and fix a range of bugs
|
2023-05-19 11:40:17 -07:00 |
convolutional
|
Add mobilenet and efficientnet models.
|
2023-05-22 16:39:24 -07:00 |
core
|
Add grad_cam, VAE examples
|
2023-05-29 09:11:23 -07:00 |
merging
|
Add DenseNet
|
2023-05-22 16:51:41 -07:00 |
pooling
|
Fix various issues
|
2023-05-13 20:17:18 -07:00 |
preprocessing
|
Fix test
|
2023-05-22 16:56:24 -07:00 |
regularization
|
Allow dynamic shapes for JAX.
|
2023-05-20 19:01:01 -07:00 |
rnn
|
Minor fixes
|
2023-05-28 14:35:22 -07:00 |
input_spec.py
|
Add some numpy ops (#1)
|
2023-04-12 11:31:58 -07:00 |
layer_test.py
|
Add layers.MultiHeadAttention (#169)
|
2023-05-16 18:22:42 -07:00 |
layer.py
|
Add support for legacy input_shape/dim use case
|
2023-05-27 10:08:09 -07:00 |