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,156Updated last year
• allenai / allennlp
  An open-source NLP research library, built on PyTorch.
  ☆11,850Updated 2 years ago
• facebookresearch / pytext
  A natural language modeling framework based on PyTorch
  ☆6,325Updated 2 years ago
• zihangdai / xlnet
  XLNet: Generalized Autoregressive Pretraining for Language Understanding
  ☆6,186Updated 2 years ago
• horovod / horovod
  Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
  ☆14,494Updated last month
• google / sentencepiece
  Unsupervised text tokenizer for Neural Network-based text generation.
  ☆10,942Updated last month
• microsoft / nlp-recipes
  Natural Language Processing Best Practices & Examples
  ☆6,412Updated 2 years ago
• stanfordnlp / stanza
  Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages
  ☆7,475Updated this week
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,575Updated this week
• flairNLP / flair
  A very simple framework for state-of-the-art Natural Language Processing (NLP)
  ☆14,182Updated this week
• facebookresearch / ParlAI
  A framework for training and evaluating AI models on a variety of openly available dialogue datasets.
  ☆10,531Updated last year
• facebookresearch / dlrm
  An implementation of a deep learning recommendation model (DLRM)
  ☆3,885Updated this week
• facebookresearch / PyTorch-BigGraph
  Generate embeddings from large-scale graph-structured data.
  ☆3,413Updated last year
• codertimo / BERT-pytorch
  Google AI 2018 BERT pytorch implementation
  ☆6,410Updated last year
• HIPS / autograd
  Efficiently computes derivatives of NumPy code.
  ☆7,276Updated this week
• pyro-ppl / pyro
  Deep universal probabilistic programming with Python and PyTorch
  ☆8,773Updated last month
• google-research / albert
  ALBERT: A Lite BERT for Self-supervised Learning of Language Representations
  ☆3,272Updated 2 years ago
• facebookresearch / mmf
  A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)
  ☆5,565Updated last month
• keras-team / autokeras
  AutoML library for deep learning
  ☆9,235Updated 5 months ago
• google-research / bert
  TensorFlow code and pre-trained models for BERT
  ☆39,176Updated 10 months ago
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,943Updated last month
• skorch-dev / skorch
  A scikit-learn compatible neural network library that wraps PyTorch
  ☆6,032Updated last month
• jessevig / bertviz
  BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)
  ☆7,430Updated last year
• OpenNMT / OpenNMT-py
  Open Source Neural Machine Translation and (Large) Language Models in PyTorch
  ☆6,881Updated 2 months ago
• facebookresearch / StarSpace
  Learning embeddings for classification, retrieval and ranking.
  ☆3,953Updated 2 years ago
• piskvorky / gensim
  Topic Modelling for Humans
  ☆16,034Updated 3 months ago
• facebookresearch / faiss
  A library for efficient similarity search and clustering of dense vectors.
  ☆35,219Updated this week
• facebookresearch / ReAgent
  A platform for Reasoning systems (Reinforcement Learning, Contextual Bandits, etc.)
  ☆3,623Updated 2 months ago
• optuna / optuna
  A hyperparameter optimization framework
  ☆12,009Updated this week
• huggingface / tokenizers
  💥 Fast State-of-the-Art Tokenizers optimized for Research and Production
  ☆9,726Updated this week