Alternatives and similar repositories for trax

Users that are interested in trax are comparing it to the libraries listed below

• tensorflow / tensor2tensor
  Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
  ☆16,466Updated 2 years ago
• allenai / allennlp
  An open-source NLP research library, built on PyTorch.
  ☆11,878Updated 2 years ago
• facebookresearch / pytext
  A natural language modeling framework based on PyTorch
  ☆6,323Updated 2 years ago
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,793Updated this week
• horovod / horovod
  Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
  ☆14,592Updated last month
• google-research / text-to-text-transfer-transformer
  Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"
  ☆6,421Updated 4 months ago
• google / dopamine
  Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.
  ☆10,803Updated 10 months ago
• huggingface / tokenizers
  💥 Fast State-of-the-Art Tokenizers optimized for Research and Production
  ☆10,060Updated 2 weeks ago
• google-deepmind / dm-haiku
  JAX-based neural network library
  ☆3,090Updated last week
• google-research / google-research
  Google Research
  ☆36,326Updated last week
• jax-ml / jax
  Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
  ☆33,428Updated this week
• google-research / albert
  ALBERT: A Lite BERT for Self-supervised Learning of Language Representations
  ☆3,274Updated 2 years ago
• google / sentencepiece
  Unsupervised text tokenizer for Neural Network-based text generation.
  ☆11,246Updated last week
• pytorch / ignite
  High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
  ☆4,698Updated this week
• zihangdai / xlnet
  XLNet: Generalized Autoregressive Pretraining for Language Understanding
  ☆6,180Updated 2 years ago
• activeloopai / deeplake
  Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangChain. Store, query, version, & visualize any AI data. Stre…
  ☆8,810Updated last week
• PAIR-code / lit
  The Learning Interpretability Tool: Interactively analyze ML models to understand their behavior in an extensible and framework agnostic …
  ☆3,589Updated 3 weeks ago
• pyro-ppl / pyro
  Deep universal probabilistic programming with Python and PyTorch
  ☆8,853Updated 2 months ago
• microsoft / nlp-recipes
  Natural Language Processing Best Practices & Examples
  ☆6,428Updated 3 years ago
• flairNLP / flair
  A very simple framework for state-of-the-art Natural Language Processing (NLP)
  ☆14,284Updated 3 weeks ago
• facebookresearch / mmf
  A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)
  ☆5,586Updated 4 months ago
• skorch-dev / skorch
  A scikit-learn compatible neural network library that wraps PyTorch
  ☆6,113Updated last month
• cortexlabs / cortex
  Production infrastructure for machine learning at scale
  ☆8,034Updated last year
• huggingface / awesome-papers
  Papers & presentation materials from Hugging Face's internal science day
  ☆2,048Updated 4 years ago
• facebookresearch / nevergrad
  A Python toolbox for performing gradient-free optimization
  ☆4,114Updated 3 months ago
• facebookresearch / hiplot
  HiPlot makes understanding high dimensional data easy
  ☆2,801Updated last year
• optuna / optuna
  A hyperparameter optimization framework
  ☆12,624Updated this week
• tensorflow / mesh
  Mesh TensorFlow: Model Parallelism Made Easier
  ☆1,615Updated last year
• huggingface / knockknock
  🚪✊Knock Knock: Get notified when your training ends with only two additional lines of code
  ☆2,817Updated 2 years ago
• HIPS / autograd
  Efficiently computes derivatives of NumPy code.
  ☆7,352Updated this week