├── .github └── workflows │ ├── create-release.yml │ ├── publish-package.yml │ └── run-tests.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CHANGELOG.md ├── LICENSE ├── README.md ├── docs ├── api │ ├── Apply.md │ ├── ArrayLike.md │ ├── BatchNorm.md │ ├── BatchStat.md │ ├── Cache.md │ ├── Compact.md │ ├── Conv.md │ ├── ConvTranspose.md │ ├── Copy.md │ ├── Dropout.md │ ├── Embed.md │ ├── Extensions.md │ ├── FieldInfo.md │ ├── FieldMetadata.md │ ├── Filter.md │ ├── Filters.md │ ├── Flatten.md │ ├── FlattenMode.md │ ├── FlaxModule.md │ ├── GroupNorm.md │ ├── HaikuModule.md │ ├── Hashable.md │ ├── Initializer.md │ ├── Inputs.md │ ├── KeySeq.md │ ├── KindMixin.md │ ├── Lambda.md │ ├── LayerNorm.md │ ├── Linear.md │ ├── Log.md │ ├── Loss.md │ ├── LossAndLogs.md │ ├── LossLog.md │ ├── MISSING.md │ ├── MLP.md │ ├── Map.md │ ├── Merge.md │ ├── Metric.md │ ├── MetricLog.md │ ├── MetricState.md │ ├── Missing.md │ ├── ModelState.md │ ├── Module.md │ ├── ModuleMeta.md │ ├── NOTHING.md │ ├── Named.md │ ├── Nothing.md │ ├── Opaque.md │ ├── OpaquePredicate.md │ ├── OptState.md │ ├── Optimizer.md │ ├── Parameter.md │ ├── Repr.md │ ├── Rng.md │ ├── Sequential.md │ ├── State.md │ ├── ToDict.md │ ├── ToString.md │ ├── Tree.md │ ├── TreeMeta.md │ ├── TreePart.md │ ├── Treex.md │ ├── add_field_info.md │ ├── apply.md │ ├── compact.md │ ├── compact_module.md │ ├── copy.md │ ├── field.md │ ├── filter.md │ ├── flatten_mode.md │ ├── in_compact.md │ ├── losses │ │ ├── CosineSimilarity.md │ │ ├── Crossentropy.md │ │ ├── Huber.md │ │ ├── Loss.md │ │ ├── MeanAbsoluteError.md │ │ ├── MeanAbsolutePercentageError.md │ │ ├── MeanSquaredError.md │ │ ├── MeanSquaredLogarithmicError.md │ │ ├── Reduction.md │ │ ├── cosine_similarity.md │ │ ├── crossentropy.md │ │ ├── huber.md │ │ ├── mean_absolute_error.md │ │ ├── mean_absolute_percentage_error.md │ │ ├── mean_squared_error.md │ │ └── mean_squared_logarithmic_error.md │ ├── map.md │ ├── merge.md │ ├── metrics │ │ ├── Accuracy.md │ │ ├── AuxLosses.md │ │ ├── AuxMetrics.md │ │ ├── LossAndLogs.md │ │ ├── Losses.md │ │ ├── MAE.md │ │ ├── MSE.md │ │ ├── Mean.md │ │ ├── MeanAbsoluteError.md │ │ ├── MeanSquareError.md │ │ ├── Metric.md │ │ ├── Metrics.md │ │ ├── Reduce.md │ │ └── Reduction.md │ ├── next_key.md │ ├── nn │ │ ├── BatchNorm.md │ │ ├── Conv.md │ │ ├── ConvTranspose.md │ │ ├── Dropout.md │ │ ├── Embed.md │ │ ├── Flatten.md │ │ ├── FlaxModule.md │ │ ├── GroupNorm.md │ │ ├── HaikuModule.md │ │ ├── Lambda.md │ │ ├── LayerNorm.md │ │ ├── Linear.md │ │ ├── MLP.md │ │ ├── Sequential.md │ │ └── sequence.md │ ├── node.md │ ├── preserve_state.md │ ├── regularizers │ │ ├── L1.md │ │ ├── L1L2.md │ │ └── L2.md │ ├── rng_key.md │ ├── sequence.md │ ├── static.md │ ├── to_dict.md │ └── to_string.md ├── index.md └── user-guide │ ├── api │ ├── filter.md │ ├── map.md │ └── merge.md │ ├── defining-modules.md │ ├── freezing-modules.md │ ├── initialization.md │ ├── intro.md │ ├── optimizer.md │ ├── pytrees.md │ ├── state-management.md │ └── training-state.md ├── examples ├── cnn.py ├── cnn_distributed.py ├── cnn_flax.py ├── cnn_haiku.py ├── cnn_model.py ├── cnn_model_distributed.py ├── linear_regression.py ├── mlp.py ├── noisy_linear │ ├── noisy_linear.ipynb │ └── noisy_linear.py ├── sdg.py ├── vae.py └── xor.py ├── images └── types.png ├── mkdocs.yml ├── poetry.lock ├── pyproject.toml ├── requirements.txt ├── scripts ├── deploy-docs.sh ├── get-coverage.sh ├── install-remote.sh ├── run-docs.sh ├── test-examples.sh ├── update_docs.py └── update_version.py ├── tests ├── losses │ ├── cosine_similarity_test.py │ ├── crossentropy_test.py │ ├── huber_test.py │ ├── loss_test.py │ ├── mean_absolute_error_test.py │ ├── mean_absolute_percentage_error_test.py │ ├── mean_squared_error_test.py │ └── mean_squared_logarithmic_error_test.py ├── metrics │ ├── test_accuracy.py │ ├── test_loss_and_logs.py │ ├── test_losses.py │ ├── test_mean_absolute_error.py │ ├── test_mean_square_error.py │ ├── test_metric.py │ └── test_metrics.py ├── nn │ ├── test_conv.py │ ├── test_dropout.py │ ├── test_embed.py │ ├── test_flax_module.py │ ├── test_linear.py │ ├── test_mlp.py │ ├── test_norm.py │ ├── test_recurrent.py │ └── test_sequential.py ├── test_optimizer.py ├── test_rng_seq.py └── test_treex.py └── treex ├── __init__.py ├── contexts.py ├── key_seq.py ├── losses ├── __init__.py ├── cosine_similarity.py ├── crossentropy.py ├── huber.py ├── loss.py ├── mean_absolute_error.py ├── mean_absolute_percentage_error.py ├── mean_squared_error.py └── mean_squared_logarithmic_error.py ├── metrics ├── __init__.py ├── accuracy.py ├── loss_and_logs.py ├── losses.py ├── mean.py ├── mean_absolute_error.py ├── mean_square_error.py ├── metric.py ├── metrics.py ├── old │ └── accuracy.py ├── reduce.py ├── tm_port │ ├── __init__.py │ ├── classification │ │ ├── __init__.py │ │ ├── accuracy.py │ │ └── stat_scores.py │ ├── functional │ │ ├── __init__.py │ │ └── classification │ │ │ ├── __init__.py │ │ │ ├── accuracy.py │ │ │ └── stat_scores.py │ └── utilities │ │ ├── __init__.py │ │ ├── checks.py │ │ ├── data.py │ │ ├── distributed.py │ │ ├── enums.py │ │ ├── exceptions.py │ │ ├── imports.py │ │ └── prints.py └── utils.py ├── module.py ├── nn ├── __init__.py ├── conv.py ├── dropout.py ├── embed.py ├── flatten.py ├── flax_module.py ├── haiku_module.py ├── linear.py ├── mlp.py ├── norm.py ├── recurrent.py └── sequential.py ├── optimizer.py ├── regularizers ├── __init__.py └── l1l2.py ├── treex.py ├── types.py └── utils.py /.github/workflows/create-release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/.github/workflows/create-release.yml -------------------------------------------------------------------------------- /.github/workflows/publish-package.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/.github/workflows/publish-package.yml -------------------------------------------------------------------------------- /.github/workflows/run-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/.github/workflows/run-tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/README.md -------------------------------------------------------------------------------- /docs/api/Apply.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Apply.md -------------------------------------------------------------------------------- /docs/api/ArrayLike.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/ArrayLike.md -------------------------------------------------------------------------------- /docs/api/BatchNorm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/BatchNorm.md -------------------------------------------------------------------------------- /docs/api/BatchStat.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/BatchStat.md -------------------------------------------------------------------------------- /docs/api/Cache.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Cache.md -------------------------------------------------------------------------------- /docs/api/Compact.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Compact.md -------------------------------------------------------------------------------- /docs/api/Conv.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Conv.md -------------------------------------------------------------------------------- /docs/api/ConvTranspose.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/ConvTranspose.md -------------------------------------------------------------------------------- /docs/api/Copy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Copy.md -------------------------------------------------------------------------------- /docs/api/Dropout.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Dropout.md -------------------------------------------------------------------------------- /docs/api/Embed.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Embed.md -------------------------------------------------------------------------------- /docs/api/Extensions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Extensions.md -------------------------------------------------------------------------------- /docs/api/FieldInfo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/FieldInfo.md -------------------------------------------------------------------------------- /docs/api/FieldMetadata.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/FieldMetadata.md -------------------------------------------------------------------------------- /docs/api/Filter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Filter.md -------------------------------------------------------------------------------- /docs/api/Filters.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Filters.md -------------------------------------------------------------------------------- /docs/api/Flatten.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Flatten.md -------------------------------------------------------------------------------- /docs/api/FlattenMode.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/FlattenMode.md -------------------------------------------------------------------------------- /docs/api/FlaxModule.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/FlaxModule.md -------------------------------------------------------------------------------- /docs/api/GroupNorm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/GroupNorm.md -------------------------------------------------------------------------------- /docs/api/HaikuModule.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/HaikuModule.md -------------------------------------------------------------------------------- /docs/api/Hashable.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Hashable.md -------------------------------------------------------------------------------- /docs/api/Initializer.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Initializer.md -------------------------------------------------------------------------------- /docs/api/Inputs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Inputs.md -------------------------------------------------------------------------------- /docs/api/KeySeq.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/KeySeq.md -------------------------------------------------------------------------------- /docs/api/KindMixin.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/KindMixin.md -------------------------------------------------------------------------------- /docs/api/Lambda.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Lambda.md -------------------------------------------------------------------------------- /docs/api/LayerNorm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/LayerNorm.md -------------------------------------------------------------------------------- /docs/api/Linear.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Linear.md -------------------------------------------------------------------------------- /docs/api/Log.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Log.md -------------------------------------------------------------------------------- /docs/api/Loss.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Loss.md -------------------------------------------------------------------------------- /docs/api/LossAndLogs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/LossAndLogs.md -------------------------------------------------------------------------------- /docs/api/LossLog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/LossLog.md -------------------------------------------------------------------------------- /docs/api/MISSING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/MISSING.md -------------------------------------------------------------------------------- /docs/api/MLP.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/MLP.md -------------------------------------------------------------------------------- /docs/api/Map.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Map.md -------------------------------------------------------------------------------- /docs/api/Merge.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Merge.md -------------------------------------------------------------------------------- /docs/api/Metric.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Metric.md -------------------------------------------------------------------------------- /docs/api/MetricLog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/MetricLog.md -------------------------------------------------------------------------------- /docs/api/MetricState.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/MetricState.md -------------------------------------------------------------------------------- /docs/api/Missing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Missing.md -------------------------------------------------------------------------------- /docs/api/ModelState.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/ModelState.md -------------------------------------------------------------------------------- /docs/api/Module.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Module.md -------------------------------------------------------------------------------- /docs/api/ModuleMeta.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/ModuleMeta.md -------------------------------------------------------------------------------- /docs/api/NOTHING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/NOTHING.md -------------------------------------------------------------------------------- /docs/api/Named.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Named.md -------------------------------------------------------------------------------- /docs/api/Nothing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Nothing.md -------------------------------------------------------------------------------- /docs/api/Opaque.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Opaque.md -------------------------------------------------------------------------------- /docs/api/OpaquePredicate.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/OpaquePredicate.md -------------------------------------------------------------------------------- /docs/api/OptState.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/OptState.md -------------------------------------------------------------------------------- /docs/api/Optimizer.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Optimizer.md -------------------------------------------------------------------------------- /docs/api/Parameter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Parameter.md -------------------------------------------------------------------------------- /docs/api/Repr.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Repr.md -------------------------------------------------------------------------------- /docs/api/Rng.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Rng.md -------------------------------------------------------------------------------- /docs/api/Sequential.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Sequential.md -------------------------------------------------------------------------------- /docs/api/State.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/State.md -------------------------------------------------------------------------------- /docs/api/ToDict.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/ToDict.md -------------------------------------------------------------------------------- /docs/api/ToString.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/ToString.md -------------------------------------------------------------------------------- /docs/api/Tree.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Tree.md -------------------------------------------------------------------------------- /docs/api/TreeMeta.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/TreeMeta.md -------------------------------------------------------------------------------- /docs/api/TreePart.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/TreePart.md -------------------------------------------------------------------------------- /docs/api/Treex.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/Treex.md -------------------------------------------------------------------------------- /docs/api/add_field_info.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/add_field_info.md -------------------------------------------------------------------------------- /docs/api/apply.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/apply.md -------------------------------------------------------------------------------- /docs/api/compact.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/compact.md -------------------------------------------------------------------------------- /docs/api/compact_module.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/compact_module.md -------------------------------------------------------------------------------- /docs/api/copy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/copy.md -------------------------------------------------------------------------------- /docs/api/field.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/field.md -------------------------------------------------------------------------------- /docs/api/filter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/filter.md -------------------------------------------------------------------------------- /docs/api/flatten_mode.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/flatten_mode.md -------------------------------------------------------------------------------- /docs/api/in_compact.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/in_compact.md -------------------------------------------------------------------------------- /docs/api/losses/CosineSimilarity.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/CosineSimilarity.md -------------------------------------------------------------------------------- /docs/api/losses/Crossentropy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/Crossentropy.md -------------------------------------------------------------------------------- /docs/api/losses/Huber.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/Huber.md -------------------------------------------------------------------------------- /docs/api/losses/Loss.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/Loss.md -------------------------------------------------------------------------------- /docs/api/losses/MeanAbsoluteError.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/MeanAbsoluteError.md -------------------------------------------------------------------------------- /docs/api/losses/MeanAbsolutePercentageError.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/MeanAbsolutePercentageError.md -------------------------------------------------------------------------------- /docs/api/losses/MeanSquaredError.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/MeanSquaredError.md -------------------------------------------------------------------------------- /docs/api/losses/MeanSquaredLogarithmicError.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/MeanSquaredLogarithmicError.md -------------------------------------------------------------------------------- /docs/api/losses/Reduction.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/Reduction.md -------------------------------------------------------------------------------- /docs/api/losses/cosine_similarity.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/cosine_similarity.md -------------------------------------------------------------------------------- /docs/api/losses/crossentropy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/crossentropy.md -------------------------------------------------------------------------------- /docs/api/losses/huber.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/huber.md -------------------------------------------------------------------------------- /docs/api/losses/mean_absolute_error.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/mean_absolute_error.md -------------------------------------------------------------------------------- /docs/api/losses/mean_absolute_percentage_error.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/mean_absolute_percentage_error.md -------------------------------------------------------------------------------- /docs/api/losses/mean_squared_error.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/mean_squared_error.md -------------------------------------------------------------------------------- /docs/api/losses/mean_squared_logarithmic_error.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/losses/mean_squared_logarithmic_error.md -------------------------------------------------------------------------------- /docs/api/map.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/map.md -------------------------------------------------------------------------------- /docs/api/merge.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/merge.md -------------------------------------------------------------------------------- /docs/api/metrics/Accuracy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/Accuracy.md -------------------------------------------------------------------------------- /docs/api/metrics/AuxLosses.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/AuxLosses.md -------------------------------------------------------------------------------- /docs/api/metrics/AuxMetrics.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/AuxMetrics.md -------------------------------------------------------------------------------- /docs/api/metrics/LossAndLogs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/LossAndLogs.md -------------------------------------------------------------------------------- /docs/api/metrics/Losses.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/Losses.md -------------------------------------------------------------------------------- /docs/api/metrics/MAE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/MAE.md -------------------------------------------------------------------------------- /docs/api/metrics/MSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/MSE.md -------------------------------------------------------------------------------- /docs/api/metrics/Mean.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/Mean.md -------------------------------------------------------------------------------- /docs/api/metrics/MeanAbsoluteError.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/MeanAbsoluteError.md -------------------------------------------------------------------------------- /docs/api/metrics/MeanSquareError.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/MeanSquareError.md -------------------------------------------------------------------------------- /docs/api/metrics/Metric.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/Metric.md -------------------------------------------------------------------------------- /docs/api/metrics/Metrics.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/Metrics.md -------------------------------------------------------------------------------- /docs/api/metrics/Reduce.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/Reduce.md -------------------------------------------------------------------------------- /docs/api/metrics/Reduction.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/metrics/Reduction.md -------------------------------------------------------------------------------- /docs/api/next_key.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/next_key.md -------------------------------------------------------------------------------- /docs/api/nn/BatchNorm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/BatchNorm.md -------------------------------------------------------------------------------- /docs/api/nn/Conv.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/Conv.md -------------------------------------------------------------------------------- /docs/api/nn/ConvTranspose.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/ConvTranspose.md -------------------------------------------------------------------------------- /docs/api/nn/Dropout.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/Dropout.md -------------------------------------------------------------------------------- /docs/api/nn/Embed.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/Embed.md -------------------------------------------------------------------------------- /docs/api/nn/Flatten.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/Flatten.md -------------------------------------------------------------------------------- /docs/api/nn/FlaxModule.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/FlaxModule.md -------------------------------------------------------------------------------- /docs/api/nn/GroupNorm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/GroupNorm.md -------------------------------------------------------------------------------- /docs/api/nn/HaikuModule.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/HaikuModule.md -------------------------------------------------------------------------------- /docs/api/nn/Lambda.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/Lambda.md -------------------------------------------------------------------------------- /docs/api/nn/LayerNorm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/LayerNorm.md -------------------------------------------------------------------------------- /docs/api/nn/Linear.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/Linear.md -------------------------------------------------------------------------------- /docs/api/nn/MLP.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/MLP.md -------------------------------------------------------------------------------- /docs/api/nn/Sequential.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/Sequential.md -------------------------------------------------------------------------------- /docs/api/nn/sequence.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/nn/sequence.md -------------------------------------------------------------------------------- /docs/api/node.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/node.md -------------------------------------------------------------------------------- /docs/api/preserve_state.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/preserve_state.md -------------------------------------------------------------------------------- /docs/api/regularizers/L1.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/regularizers/L1.md -------------------------------------------------------------------------------- /docs/api/regularizers/L1L2.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/regularizers/L1L2.md -------------------------------------------------------------------------------- /docs/api/regularizers/L2.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/regularizers/L2.md -------------------------------------------------------------------------------- /docs/api/rng_key.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/rng_key.md -------------------------------------------------------------------------------- /docs/api/sequence.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/sequence.md -------------------------------------------------------------------------------- /docs/api/static.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/static.md -------------------------------------------------------------------------------- /docs/api/to_dict.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/to_dict.md -------------------------------------------------------------------------------- /docs/api/to_string.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/api/to_string.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/user-guide/api/filter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/user-guide/api/filter.md -------------------------------------------------------------------------------- /docs/user-guide/api/map.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/user-guide/api/map.md -------------------------------------------------------------------------------- /docs/user-guide/api/merge.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/user-guide/api/merge.md -------------------------------------------------------------------------------- /docs/user-guide/defining-modules.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/user-guide/defining-modules.md -------------------------------------------------------------------------------- /docs/user-guide/freezing-modules.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/user-guide/freezing-modules.md -------------------------------------------------------------------------------- /docs/user-guide/initialization.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/user-guide/initialization.md -------------------------------------------------------------------------------- /docs/user-guide/intro.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/user-guide/intro.md -------------------------------------------------------------------------------- /docs/user-guide/optimizer.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/user-guide/optimizer.md -------------------------------------------------------------------------------- /docs/user-guide/pytrees.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/user-guide/state-management.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/user-guide/state-management.md -------------------------------------------------------------------------------- /docs/user-guide/training-state.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/docs/user-guide/training-state.md -------------------------------------------------------------------------------- /examples/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/cnn.py -------------------------------------------------------------------------------- /examples/cnn_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/cnn_distributed.py -------------------------------------------------------------------------------- /examples/cnn_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/cnn_flax.py -------------------------------------------------------------------------------- /examples/cnn_haiku.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/cnn_haiku.py -------------------------------------------------------------------------------- /examples/cnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/cnn_model.py -------------------------------------------------------------------------------- /examples/cnn_model_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/cnn_model_distributed.py -------------------------------------------------------------------------------- /examples/linear_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/linear_regression.py -------------------------------------------------------------------------------- /examples/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/mlp.py -------------------------------------------------------------------------------- /examples/noisy_linear/noisy_linear.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/noisy_linear/noisy_linear.ipynb -------------------------------------------------------------------------------- /examples/noisy_linear/noisy_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/noisy_linear/noisy_linear.py -------------------------------------------------------------------------------- /examples/sdg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/sdg.py -------------------------------------------------------------------------------- /examples/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/vae.py -------------------------------------------------------------------------------- /examples/xor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/examples/xor.py -------------------------------------------------------------------------------- /images/types.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/images/types.png -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/deploy-docs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/scripts/deploy-docs.sh -------------------------------------------------------------------------------- /scripts/get-coverage.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/scripts/get-coverage.sh -------------------------------------------------------------------------------- /scripts/install-remote.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/scripts/install-remote.sh -------------------------------------------------------------------------------- /scripts/run-docs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/scripts/run-docs.sh -------------------------------------------------------------------------------- /scripts/test-examples.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/scripts/test-examples.sh -------------------------------------------------------------------------------- /scripts/update_docs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/scripts/update_docs.py -------------------------------------------------------------------------------- /scripts/update_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/scripts/update_version.py -------------------------------------------------------------------------------- /tests/losses/cosine_similarity_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/losses/cosine_similarity_test.py -------------------------------------------------------------------------------- /tests/losses/crossentropy_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/losses/crossentropy_test.py -------------------------------------------------------------------------------- /tests/losses/huber_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/losses/huber_test.py -------------------------------------------------------------------------------- /tests/losses/loss_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/losses/loss_test.py -------------------------------------------------------------------------------- /tests/losses/mean_absolute_error_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/losses/mean_absolute_error_test.py -------------------------------------------------------------------------------- /tests/losses/mean_absolute_percentage_error_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/losses/mean_absolute_percentage_error_test.py -------------------------------------------------------------------------------- /tests/losses/mean_squared_error_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/losses/mean_squared_error_test.py -------------------------------------------------------------------------------- /tests/losses/mean_squared_logarithmic_error_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/losses/mean_squared_logarithmic_error_test.py -------------------------------------------------------------------------------- /tests/metrics/test_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/metrics/test_accuracy.py -------------------------------------------------------------------------------- /tests/metrics/test_loss_and_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/metrics/test_loss_and_logs.py -------------------------------------------------------------------------------- /tests/metrics/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/metrics/test_losses.py -------------------------------------------------------------------------------- /tests/metrics/test_mean_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/metrics/test_mean_absolute_error.py -------------------------------------------------------------------------------- /tests/metrics/test_mean_square_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/metrics/test_mean_square_error.py -------------------------------------------------------------------------------- /tests/metrics/test_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/metrics/test_metric.py -------------------------------------------------------------------------------- /tests/metrics/test_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/metrics/test_metrics.py -------------------------------------------------------------------------------- /tests/nn/test_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/nn/test_conv.py -------------------------------------------------------------------------------- /tests/nn/test_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/nn/test_dropout.py -------------------------------------------------------------------------------- /tests/nn/test_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/nn/test_embed.py -------------------------------------------------------------------------------- /tests/nn/test_flax_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/nn/test_flax_module.py -------------------------------------------------------------------------------- /tests/nn/test_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/nn/test_linear.py -------------------------------------------------------------------------------- /tests/nn/test_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/nn/test_mlp.py -------------------------------------------------------------------------------- /tests/nn/test_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/nn/test_norm.py -------------------------------------------------------------------------------- /tests/nn/test_recurrent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/nn/test_recurrent.py -------------------------------------------------------------------------------- /tests/nn/test_sequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/nn/test_sequential.py -------------------------------------------------------------------------------- /tests/test_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/test_optimizer.py -------------------------------------------------------------------------------- /tests/test_rng_seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/test_rng_seq.py -------------------------------------------------------------------------------- /tests/test_treex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/tests/test_treex.py -------------------------------------------------------------------------------- /treex/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/__init__.py -------------------------------------------------------------------------------- /treex/contexts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/contexts.py -------------------------------------------------------------------------------- /treex/key_seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/key_seq.py -------------------------------------------------------------------------------- /treex/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/losses/__init__.py -------------------------------------------------------------------------------- /treex/losses/cosine_similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/losses/cosine_similarity.py -------------------------------------------------------------------------------- /treex/losses/crossentropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/losses/crossentropy.py -------------------------------------------------------------------------------- /treex/losses/huber.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/losses/huber.py -------------------------------------------------------------------------------- /treex/losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/losses/loss.py -------------------------------------------------------------------------------- /treex/losses/mean_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/losses/mean_absolute_error.py -------------------------------------------------------------------------------- /treex/losses/mean_absolute_percentage_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/losses/mean_absolute_percentage_error.py -------------------------------------------------------------------------------- /treex/losses/mean_squared_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/losses/mean_squared_error.py -------------------------------------------------------------------------------- /treex/losses/mean_squared_logarithmic_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/losses/mean_squared_logarithmic_error.py -------------------------------------------------------------------------------- /treex/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/__init__.py -------------------------------------------------------------------------------- /treex/metrics/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/accuracy.py -------------------------------------------------------------------------------- /treex/metrics/loss_and_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/loss_and_logs.py -------------------------------------------------------------------------------- /treex/metrics/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/losses.py -------------------------------------------------------------------------------- /treex/metrics/mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/mean.py -------------------------------------------------------------------------------- /treex/metrics/mean_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/mean_absolute_error.py -------------------------------------------------------------------------------- /treex/metrics/mean_square_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/mean_square_error.py -------------------------------------------------------------------------------- /treex/metrics/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/metric.py -------------------------------------------------------------------------------- /treex/metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/metrics.py -------------------------------------------------------------------------------- /treex/metrics/old/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/old/accuracy.py -------------------------------------------------------------------------------- /treex/metrics/reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/reduce.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/__init__.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/classification/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/classification/__init__.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/classification/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/classification/accuracy.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/classification/stat_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/classification/stat_scores.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/functional/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /treex/metrics/tm_port/functional/classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /treex/metrics/tm_port/functional/classification/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/functional/classification/accuracy.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/functional/classification/stat_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/functional/classification/stat_scores.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/utilities/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/utilities/__init__.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/utilities/checks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/utilities/checks.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/utilities/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/utilities/data.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/utilities/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/utilities/distributed.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/utilities/enums.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/utilities/enums.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/utilities/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/utilities/exceptions.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/utilities/imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/utilities/imports.py -------------------------------------------------------------------------------- /treex/metrics/tm_port/utilities/prints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/tm_port/utilities/prints.py -------------------------------------------------------------------------------- /treex/metrics/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/metrics/utils.py -------------------------------------------------------------------------------- /treex/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/module.py -------------------------------------------------------------------------------- /treex/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/__init__.py -------------------------------------------------------------------------------- /treex/nn/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/conv.py -------------------------------------------------------------------------------- /treex/nn/dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/dropout.py -------------------------------------------------------------------------------- /treex/nn/embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/embed.py -------------------------------------------------------------------------------- /treex/nn/flatten.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/flatten.py -------------------------------------------------------------------------------- /treex/nn/flax_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/flax_module.py -------------------------------------------------------------------------------- /treex/nn/haiku_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/haiku_module.py -------------------------------------------------------------------------------- /treex/nn/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/linear.py -------------------------------------------------------------------------------- /treex/nn/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/mlp.py -------------------------------------------------------------------------------- /treex/nn/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/norm.py -------------------------------------------------------------------------------- /treex/nn/recurrent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/recurrent.py -------------------------------------------------------------------------------- /treex/nn/sequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/nn/sequential.py -------------------------------------------------------------------------------- /treex/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/optimizer.py -------------------------------------------------------------------------------- /treex/regularizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/regularizers/__init__.py -------------------------------------------------------------------------------- /treex/regularizers/l1l2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/regularizers/l1l2.py -------------------------------------------------------------------------------- /treex/treex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/treex.py -------------------------------------------------------------------------------- /treex/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/types.py -------------------------------------------------------------------------------- /treex/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgarciae/treex/HEAD/treex/utils.py --------------------------------------------------------------------------------