All modules for which code is available

• nerva_jax.activation_functions
• nerva_jax.datasets
• nerva_jax.layers
• nerva_jax.learning_rate
• nerva_jax.loss_functions
• nerva_jax.matrix_operations
• nerva_jax.multilayer_perceptron
• nerva_jax.optimizers
• nerva_jax.softmax_functions
• nerva_jax.training
• nerva_jax.utilities
• nerva_jax.weight_initializers