Trax library

E899039

Trax library is an open-source deep learning framework in Python focused on clear, concise code and advanced sequence modeling, developed by Google researchers.

All labels observed (1)

Label Occurrences
Trax library canonical 1

How this entity was disambiguated

Statements (40)

Predicate Object
instanceOf Python library ⓘ
deep learning framework ⓘ
open-source software ⓘ
builtOn JAX ⓘ
TensorFlow (earlier versions) ⓘ
linked to: TensorFlow
category machine learning library ⓘ
scientific computing software for Python ⓘ
developer Google ⓘ
feature automatic differentiation via JAX ⓘ
built-in example tasks ⓘ
checkpointing of models ⓘ
configurable training loops ⓘ
functional-style model definition ⓘ
focus clear and concise code ⓘ
deep learning research ⓘ
sequence modeling ⓘ
goal make advanced deep learning research easier to reproduce ⓘ
simplify experimentation with sequence models ⓘ
license Apache License 2.0 ⓘ
maintainer Google Research ⓘ
origin Google Research Brain Team ⓘ
linked to: Google Brain
programmingLanguage Python ⓘ
provides data input pipelines ⓘ
evaluation utilities ⓘ
high-level training loops ⓘ
predefined models ⓘ
predefined neural network layers ⓘ
repository https://github.com/google/trax ⓘ
supports GPU acceleration ⓘ
TPU acceleration ⓘ
convolutional neural networks ⓘ
language modeling ⓘ
machine translation ⓘ
recurrent neural networks ⓘ
reinforcement learning experiments ⓘ
text classification ⓘ
transformer models ⓘ
usedFor educational examples of deep learning ⓘ
research in natural language processing ⓘ
research in sequence-to-sequence models ⓘ

How these facts were elicited

Referenced by (1)

Full triples — surface form annotated when it differs from this entity's canonical label.

Łukasz Kaiser → developed → Trax library ⓘ
subject linked to: Lukasz Kaiser