treex
A Pytree Module system for Deep Learning in JAX
File Explorer
Download Latest Version (.zip)- create-release.yml
- publish-package.yml
- run-tests.yml
- cosine_similarity.md
- CosineSimilarity.md
- Crossentropy.md
- crossentropy.md
- Huber.md
- huber.md
- Loss.md
- mean_absolute_error.md
- mean_absolute_percentage_error.md
- mean_squared_error.md
- mean_squared_logarithmic_error.md
- MeanAbsoluteError.md
- MeanAbsolutePercentageError.md
- MeanSquaredError.md
- MeanSquaredLogarithmicError.md
- Reduction.md
- Accuracy.md
- AuxLosses.md
- AuxMetrics.md
- LossAndLogs.md
- Losses.md
- MAE.md
- Mean.md
- MeanAbsoluteError.md
- MeanSquareError.md
- Metric.md
- Metrics.md
- MSE.md
- Reduce.md
- Reduction.md
- 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
- sequence.md
- Sequential.md
- L1.md
- L1L2.md
- L2.md
- add_field_info.md
- Apply.md
- apply.md
- ArrayLike.md
- BatchNorm.md
- BatchStat.md
- Cache.md
- Compact.md
- compact.md
- compact_module.md
- Conv.md
- ConvTranspose.md
- Copy.md
- copy.md
- Dropout.md
- Embed.md
- Extensions.md
- field.md
- FieldInfo.md
- FieldMetadata.md
- Filter.md
- filter.md
- Filters.md
- Flatten.md
- flatten_mode.md
- FlattenMode.md
- FlaxModule.md
- GroupNorm.md
- HaikuModule.md
- Hashable.md
- in_compact.md
- Initializer.md
- Inputs.md
- KeySeq.md
- KindMixin.md
- Lambda.md
- LayerNorm.md
- Linear.md
- Log.md
- Loss.md
- LossAndLogs.md
- LossLog.md
- Map.md
- map.md
- Merge.md
- merge.md
- Metric.md
- MetricLog.md
- MetricState.md
- MISSING.md
- Missing.md
- MLP.md
- ModelState.md
- Module.md
- ModuleMeta.md
- Named.md
- next_key.md
- node.md
- NOTHING.md
- Nothing.md
- Opaque.md
- OpaquePredicate.md
- Optimizer.md
- OptState.md
- Parameter.md
- preserve_state.md
- Repr.md
- Rng.md
- rng_key.md
- sequence.md
- Sequential.md
- State.md
- static.md
- to_dict.md
- to_string.md
- ToDict.md
- ToString.md
- Tree.md
- TreeMeta.md
- TreePart.md
- Treex.md
- 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
- index.md
- noisy_linear.ipynb
- noisy_linear.py
- cnn.py
- cnn_distributed.py
- cnn_flax.py
- cnn_haiku.py
- cnn_model.py
- cnn_model_distributed.py
- linear_regression.py
- mlp.py
- sdg.py
- vae.py
- xor.py
- types.png
- deploy-docs.sh
- get-coverage.sh
- install-remote.sh
- run-docs.sh
- test-examples.sh
- update_docs.py
- update_version.py
- 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
- 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
- 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
- __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
- accuracy.py
- __init__.py
- accuracy.py
- stat_scores.py
- __init__.py
- accuracy.py
- stat_scores.py
- __init__.py
- __init__.py
- checks.py
- data.py
- distributed.py
- enums.py
- exceptions.py
- imports.py
- prints.py
- __init__.py
- __init__.py
- accuracy.py
- loss_and_logs.py
- losses.py
- mean.py
- mean_absolute_error.py
- mean_square_error.py
- metric.py
- metrics.py
- reduce.py
- utils.py
- __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
- __init__.py
- l1l2.py
- __init__.py
- contexts.py
- key_seq.py
- module.py
- optimizer.py
- treex.py
- types.py
- utils.py
- .gitignore
- .pre-commit-config.yaml
- build.sh
- CHANGELOG.md
- LICENSE
- mkdocs.yml
- poetry.lock
- pyproject.toml
- README.md
- requirements.txt
# Installation Guide
1. Get the code
git clone https://github.com/cgarciae/treex
Downloads the entire project code from GitHub to your computer.
cd treex
Moves into the project folder you just downloaded.
2. Official Install Script
Easy RecommendedPrerequisites
- Python 3 Python is required to use pip.
pip install treex
Installs the package published on PyPI directly โ no need to clone the source.
After installing, open a new terminal and run the program's version command (e.g. --version) to confirm it worked.
Pulled directly from this repo's README.
3. Python
EasyPrerequisites
pip install treex
Installs the package published on PyPI directly โ no need to clone the source.
If it runs without errors and prints output in the terminal, it worked.
Pulled directly from this repo's README.
// repository documentation
Was this content helpful?
(0 ratings)