equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
File Explorer
Download Latest Version (.zip)- build_docs.yml
- release.yml
- run_tests.yml
- FUNDING.yml
- source.html
- bluesky.svg
- .README.md
- custom_css.css
- favicon.png
- mathjax.js
- advanced_fields.md
- module.md
- activations.md
- attention.md
- conv.md
- dropout.md
- embedding.md
- inference.md
- linear.md
- mlp.md
- normalisation.md
- pool.md
- rnn.md
- sequential.md
- shared.md
- stateful.md
- caches.md
- debug.md
- enumerations.md
- errors.md
- manipulation.md
- pretty-printing.md
- serialisation.md
- transformations.md
- t10k-images-idx3-ubyte
- t10k-images-idx3-ubyte.gz
- t10k-labels-idx1-ubyte
- t10k-labels-idx1-ubyte.gz
- train-images-idx3-ubyte
- train-images-idx3-ubyte.gz
- train-labels-idx1-ubyte
- train-labels-idx1-ubyte.gz
- bert.ipynb
- bert_checkpoint.eqx
- deep_convolutional_gan.ipynb
- frozen_layer.ipynb
- init_apply.ipynb
- mnist.ipynb
- parallelism.ipynb
- score_based_diffusion.ipynb
- serialisation.ipynb
- stateful.ipynb
- train_rnn.ipynb
- unet.ipynb
- vision_transformer.ipynb
- .citation.md
- .htaccess
- all-of-equinox.md
- awesome-list.md
- citation.md
- faq.md
- imgs
- index.md
- pattern.md
- tricks.md
- __init__.py
- _better_abstract.py
- _field.py
- _flatten.py
- _module.py
- _prebuilt.py
- README.md
- __init__.py
- _announce_transform.py
- _backward_nan.py
- _breakpoint_if.py
- _dce.py
- _max_traces.py
- __init__.py
- bounded.py
- checkpointed.py
- common.py
- loop.py
- __init__.py
- _closure_to_pytree.py
- _finalise_jaxpr.py
- _getkey.py
- _misc.py
- _nextafter.py
- _noinline.py
- _nontraceable.py
- _omega.py
- _onnx.py
- _primitive.py
- _progress_meter.py
- _str2jax.py
- README.md
- __init__.py
- _activations.py
- _attention.py
- _batch_norm.py
- _conv.py
- _dropout.py
- _embedding.py
- _inference.py
- _linear.py
- _misc.py
- _mlp.py
- _normalisation.py
- _pool.py
- _rnn.py
- _sequential.py
- _shared.py
- _spectral_norm.py
- _stateful.py
- _weight_norm.py
- __init__.py
- _ad.py
- _caches.py
- _callback.py
- _compile_utils.py
- _config.py
- _custom_types.py
- _deprecate.py
- _doc_utils.py
- _enum.py
- _errors.py
- _eval_shape.py
- _filters.py
- _jit.py
- _make_jaxpr.py
- _misc.py
- _pretty_print.py
- _serialisation.py
- _sharding.py
- _tree.py
- _unvmap.py
- _update.py
- _vmap_pmap.py
- py.typed
- score_based_diffusion_samples.png
- __init__.py
- conftest.py
- helpers.py
- test_abstract.py
- test_ad.py
- test_caches.py
- test_callback.py
- test_checkpoint.py
- test_closure_to_pytree.py
- test_debug.py
- test_enum.py
- test_errors.py
- test_eval_shape.py
- test_filters.py
- test_finalise_jaxpr.py
- test_jit.py
- test_make_jaxpr.py
- test_misc.py
- test_module.py
- test_nn.py
- test_noinline.py
- test_nontraceable.py
- test_onnx.py
- test_pformat.py
- test_pmap.py
- test_primitive.py
- test_progress_meter.py
- test_scan.py
- test_serialisation.py
- test_sharding.py
- test_shared.py
- test_stateful.py
- test_str2jax.py
- test_tree.py
- test_update.py
- test_vmap.py
- test_while_loop.py
- .editorconfig
- .gitignore
- .pre-commit-config.yaml
- CONTRIBUTING.md
- LICENSE
- mkdocs.yml
- pyproject.toml
- README.md
# Installation Guide
1. Get the code
git clone https://github.com/patrick-kidger/equinox
Downloads the entire project code from GitHub to your computer.
cd equinox
Moves into the project folder you just downloaded.
2. Official Install Script
Easy RecommendedPrerequisites
- Python 3 Python is required to use pip.
pip install equinox
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 equinox
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)