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 NERFINISHED
TensorFlow (earlier versions) NERFINISHED
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 NERFINISHED
origin Google Research Brain Team NERFINISHED
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.

Lukasz Kaiser developed Trax library
subject surface form: Łukasz Kaiser