LICENSE
README.md
pyproject.toml
neutro/__init__.py
neutro/data.py
neutro.egg-info/PKG-INFO
neutro.egg-info/SOURCES.txt
neutro.egg-info/dependency_links.txt
neutro.egg-info/requires.txt
neutro.egg-info/top_level.txt
neutro/activations/__init__.py
neutro/activations/base.py
neutro/activations/relu.py
neutro/activations/sigmoid.py
neutro/activations/silu.py
neutro/activations/softmax.py
neutro/activations/tanh.py
neutro/callbacks/__init__.py
neutro/callbacks/base.py
neutro/callbacks/checkpoint.py
neutro/callbacks/early_stopping.py
neutro/callbacks/history.py
neutro/callbacks/lr_scheduler.py
neutro/engine/__init__.py
neutro/engine/node.py
neutro/initializers/__init__.py
neutro/initializers/base.py
neutro/initializers/constant.py
neutro/initializers/glorot.py
neutro/initializers/he.py
neutro/initializers/random.py
neutro/layers/__init__.py
neutro/layers/base.py
neutro/layers/attention/base_attention.py
neutro/layers/attention/flash_attention.py
neutro/layers/attention/gqa.py
neutro/layers/attention/kv_cache.py
neutro/layers/attention/mha.py
neutro/layers/attention/mla.py
neutro/layers/attention/mqa.py
neutro/layers/convolutional/conv1d.py
neutro/layers/convolutional/conv2d.py
neutro/layers/core/__init__.py
neutro/layers/core/activation.py
neutro/layers/core/dense.py
neutro/layers/core/dropout.py
neutro/layers/core/flatten.py
neutro/layers/core/input_layer.py
neutro/layers/core/merging.py
neutro/layers/core/moe.py
neutro/layers/core/reparameterization.py
neutro/layers/embedding/__init__.py
neutro/layers/embedding/embedding.py
neutro/layers/embedding/time_embedding.py
neutro/layers/normalization/__init__.py
neutro/layers/normalization/batchnorm.py
neutro/layers/normalization/groupnorm.py
neutro/layers/normalization/layernorm.py
neutro/layers/normalization/rmsnorm.py
neutro/layers/pooling/__init__.py
neutro/layers/pooling/global_pooling.py
neutro/layers/pooling/maxpooling2d.py
neutro/layers/pooling/upsampling2d.py
neutro/layers/recurrent/__init__.py
neutro/layers/recurrent/gru.py
neutro/layers/recurrent/lstm.py
neutro/layers/recurrent/simple_rnn.py
neutro/layers/transformer/transformer_block.py
neutro/losses/__init__.py
neutro/losses/base.py
neutro/losses/categorical_crossentropy.py
neutro/losses/mse.py
neutro/losses/sparse_categorical_crossentropy.py
neutro/losses/vae_loss.py
neutro/metrics/__init__.py
neutro/metrics/accuracy.py
neutro/metrics/base.py
neutro/metrics/f1_score.py
neutro/metrics/precision.py
neutro/metrics/recall.py
neutro/metrics/sparse_accuracy.py
neutro/models/__init__.py
neutro/models/base_model.py
neutro/models/language/__init__.py
neutro/models/language/deepseek.py
neutro/models/language/gpt.py
neutro/models/language/llama.py
neutro/models/language/qwen.py
neutro/models/moe/__init__.py
neutro/models/vision/__init__.py
neutro/models/vision/alexnet.py
neutro/models/vision/diffusion_model.py
neutro/models/vision/unet.py
neutro/models/vision/vae.py
neutro/models/vision/vgg.py
neutro/optimizers/__init__.py
neutro/optimizers/adam.py
neutro/optimizers/adamw.py
neutro/optimizers/base.py
neutro/optimizers/schedules.py
neutro/optimizers/sgd.py
neutro/preprocessing/__init__.py
neutro/preprocessing/image.py
neutro/preprocessing/sequence.py
neutro/preprocessing/text.py
neutro/tokenizers/__init__.py
neutro/tokenizers/bpe.py
neutro/tokenizers/tiktoken_compat.py
neutro/utils/__init__.py
neutro/utils/conv_utils.py
neutro/utils/data_utils.py
neutro/utils/diffusion_utils.py
neutro/utils/rope_utils.py
neutro/utils/visualization.py
tests/test_data.py
tests/test_functional_api.py
tests/test_mimo_fit.py
tests/test_preprocessing.py
tests/test_shared_transformer_block.py