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.
  ☆15,740Updated last year
• allenai / allennlp
  An open-source NLP research library, built on PyTorch.
  ☆11,782Updated 2 years ago
• google / sentencepiece
  Unsupervised text tokenizer for Neural Network-based text generation.
  ☆10,479Updated last month
• horovod / horovod
  Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
  ☆14,337Updated last month
• jax-ml / jax
  Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
  ☆30,985Updated this week
• 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,240Updated 3 months ago
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,272Updated this week
• microsoft / nlp-recipes
  Natural Language Processing Best Practices & Examples
  ☆6,386Updated 2 years ago
• OpenNMT / OpenNMT-py
  Open Source Neural Machine Translation and (Large) Language Models in PyTorch
  ☆6,803Updated last week
• facebookresearch / fairseq
  Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
  ☆30,809Updated last week
• facebookresearch / ParlAI
  A framework for training and evaluating AI models on a variety of openly available dialogue datasets.
  ☆10,496Updated last year
• stanfordnlp / stanza
  Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages
  ☆7,337Updated this week
• zihangdai / xlnet
  XLNet: Generalized Autoregressive Pretraining for Language Understanding
  ☆6,180Updated last year
• flairNLP / flair
  A very simple framework for state-of-the-art Natural Language Processing (NLP)
  ☆14,016Updated this week
• google / dopamine
  Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.
  ☆10,607Updated 2 months ago
• PAIR-code / lit
  The Learning Interpretability Tool: Interactively analyze ML models to understand their behavior in an extensible and framework agnostic …
  ☆3,510Updated 3 weeks ago
• facebookresearch / pytext
  A natural language modeling framework based on PyTorch
  ☆6,334Updated 2 years ago
• jessevig / bertviz
  BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)
  ☆7,075Updated last year
• NVIDIA / DeepLearningExamples
  State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…
  ☆13,831Updated 5 months ago
• piskvorky / gensim
  Topic Modelling for Humans
  ☆15,792Updated 3 weeks ago
• facebookresearch / mmf
  A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)
  ☆5,521Updated 2 months ago
• facebookresearch / faiss
  A library for efficient similarity search and clustering of dense vectors.
  ☆32,387Updated this week
• rapidsai / cudf
  cuDF - GPU DataFrame Library
  ☆8,597Updated this week
• NVIDIA / Megatron-LM
  Ongoing research training transformer models at scale
  ☆11,109Updated this week
• keras-team / autokeras
  AutoML library for deep learning
  ☆9,178Updated last month
• fastai / fastai
  The fastai deep learning library
  ☆26,505Updated last month
• facebookresearch / fastText
  Library for fast text representation and classification.
  ☆26,017Updated 9 months ago
• facebookresearch / PyTorch-BigGraph
  Generate embeddings from large-scale graph-structured data.
  ☆3,394Updated 10 months ago
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,655Updated this week
• facebookresearch / XLM
  PyTorch original implementation of Cross-lingual Language Model Pretraining.
  ☆2,898Updated last year