google / trax

Related projects ⓘ

Alternatives and complementary repositories for trax

• google / sentencepiece
  Unsupervised text tokenizer for Neural Network-based text generation.
  ☆10,295Updated 2 weeks ago
• allenai / allennlp
  An open-source NLP research library, built on PyTorch.
  ☆11,759Updated last year
• jax-ml / jax
  Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
  ☆30,532Updated this week
• tensorflow / tensor2tensor
  Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
  ☆15,560Updated last year
• horovod / horovod
  Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
  ☆14,265Updated 2 months ago
• zihangdai / xlnet
  XLNet: Generalized Autoregressive Pretraining for Language Understanding
  ☆6,182Updated last year
• facebookresearch / pytext
  A natural language modeling framework based on PyTorch
  ☆6,338Updated 2 years ago
• 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,178Updated 2 months ago
• snorkel-team / snorkel
  A system for quickly generating training data with weak supervision
  ☆5,812Updated 6 months ago
• huggingface / tokenizers
  💥 Fast State-of-the-Art Tokenizers optimized for Research and Production
  ☆9,052Updated this week
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,524Updated this week
• facebookresearch / mmf
  A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)
  ☆5,502Updated this week
• PAIR-code / lit
  The Learning Interpretability Tool: Interactively analyze ML models to understand their behavior in an extensible and framework agnostic …
  ☆3,492Updated 2 weeks ago
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,142Updated this week
• microsoft / tensorwatch
  Debugging, monitoring and visualization for Python Machine Learning and Data Science
  ☆3,420Updated last year
• HIPS / autograd
  Efficiently computes derivatives of NumPy code.
  ☆7,019Updated this week
• google-deepmind / sonnet
  TensorFlow-based neural network library
  ☆9,780Updated this week
• jessevig / bertviz
  BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)
  ☆6,952Updated last year
• karpathy / minGPT
  A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
  ☆20,199Updated 3 months ago
• flairNLP / flair
  A very simple framework for state-of-the-art Natural Language Processing (NLP)
  ☆13,946Updated this week
• cortexlabs / cortex
  Production infrastructure for machine learning at scale
  ☆8,021Updated 5 months ago
• stanfordnlp / stanza
  Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages
  ☆7,296Updated this week
• google / dopamine
  Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.
  ☆10,571Updated 2 weeks ago
• google-research / bert
  TensorFlow code and pre-trained models for BERT
  ☆38,213Updated 3 months ago
• microsoft / nlp-recipes
  Natural Language Processing Best Practices & Examples
  ☆6,374Updated 2 years ago
• facebookresearch / PyTorch-BigGraph
  Generate embeddings from large-scale graph-structured data.
  ☆3,387Updated 8 months ago
• facebookresearch / ParlAI
  A framework for training and evaluating AI models on a variety of openly available dialogue datasets.
  ☆10,494Updated last year
• harvardnlp / annotated-transformer
  An annotated implementation of the Transformer paper.
  ☆5,733Updated 7 months ago
• google-research / albert
  ALBERT: A Lite BERT for Self-supervised Learning of Language Representations
  ☆3,247Updated last year
• skorch-dev / skorch
  A scikit-learn compatible neural network library that wraps PyTorch
  ☆5,884Updated 2 weeks ago