以jax为后端的类似keras的框架
| activations | layers | losses | metrics | optimizers |
|---|---|---|---|---|
| Linear | Dense | binary_crossentropy | binary_accuracy | SGD |
| Softmax | Flatten | categorical_crossentropy | accuracy | SM3 |
| Relu | Dropout | mean_squared_error | categorical_accuracy | Adagrad |
| Sigmoid | Conv2D | mean_absolute_error | sparse_categorical_accuracy | Adam |
| Elu | MaxPool2D | mean_squared_logarithmic_error | cosine_similarity_accuracy | Adamax |
| LeakyRelu | AvgPool2D | hinge | top_k_categorical_accuracy | RMSProp |
| Gelu | GRU | kl_divergence | sparse_top_k_categorical_accuracy | FTRL |
| huber |