Metadata-Version: 2.1
Name: trax
Version: 1.3.4
Summary: Trax
Home-page: http://github.com/google/trax
Author: Google Inc.
Author-email: no-reply@google.com
License: Apache 2.0
Keywords: tensorflow machine learning jax
Platform: UNKNOWN
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Dist: absl-py
Requires-Dist: funcsigs
Requires-Dist: gin-config
Requires-Dist: gym
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: numpy
Requires-Dist: scipy
Requires-Dist: six
Requires-Dist: t5
Requires-Dist: tensor2tensor
Requires-Dist: tensorflow-datasets
Requires-Dist: tensorflow-text
Provides-Extra: tensorflow
Requires-Dist: tensorflow (>=1.15.0) ; extra == 'tensorflow'
Provides-Extra: tensorflow_gpu
Requires-Dist: tensorflow-gpu (>=1.15.0) ; extra == 'tensorflow_gpu'
Provides-Extra: tests
Requires-Dist: attrs ; extra == 'tests'
Requires-Dist: jupyter ; extra == 'tests'
Requires-Dist: matplotlib ; extra == 'tests'
Requires-Dist: mock ; extra == 'tests'
Requires-Dist: parameterized ; extra == 'tests'
Requires-Dist: pylint ; extra == 'tests'
Requires-Dist: pytest ; extra == 'tests'
Requires-Dist: wrapt (==1.11.*) ; extra == 'tests'

Trax helps you understand deep learning. We start with basic maths and go through layers, models, supervised and reinforcement learning. We get to advanced deep learning results, including recent papers and state-of-the-art models.


