Alternatives and similar repositories for t5-jax

Users that are interested in t5-jax are comparing it to the libraries listed below

• andylolu2 / jax-vqvae-gpt
  Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.
  ☆12Updated 2 years ago
• Sea-Snell / CALM-Dialogue
  Official code for the paper "Context-Aware Language Modeling for Goal-Oriented Dialogue Systems"
  ☆34Updated 3 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆92Updated 2 years ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆210Updated 2 years ago
• google-deepmind / enn_acme
  ☆31Updated 3 years ago
• sholtodouglas / scalingExperiments
  ☆63Updated 3 years ago
• kelechi-c / dit_flow
  DiT (training + flow matching) in Jax
  ☆11Updated last year
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆51Updated 2 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆45Updated last year
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆48Updated 6 months ago
• btnorman / First-Explore
  Repo to reproduce the First-Explore paper results
  ☆39Updated last year
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆144Updated last year
• microsoft / Intrepid
  INTeractive learning via REPresentatIon Discovery
  ☆36Updated last year
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated 2 years ago
• google-deepmind / nao_top10
  ☆19Updated 2 years ago
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆83Updated 3 years ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆132Updated last year
• google-research / precondition
  ☆31Updated last week
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆40Updated 2 months ago
• facebookresearch / motif
  Intrinsic Motivation from Artificial Intelligence Feedback
  ☆135Updated 2 years ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆92Updated last year
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Updated 4 months ago
• lucidrains / token-shift-gpt
  Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing
  ☆49Updated 4 years ago
• DHDev0 / Muzero-unplugged
  Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observ…
  ☆35Updated 7 months ago
• Sea-Snell / Implicit-Language-Q-Learning
  Official code from the paper "Offline RL for Natural Language Generation with Implicit Language Q Learning"
  ☆210Updated 2 years ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆36Updated last year
• zach-lawless / gym-wordle
  Gym environment for playing Wordle with RL agents
  ☆42Updated 3 years ago
• rovle / gpt3-in-context-fitting
  Experiments on GPT-3's ability to fit numerical models in-context.
  ☆14Updated 3 years ago
• CarperAI / Algorithm-Distillation-RLHF
  ☆35Updated 3 years ago
• lucidrains / esbn-transformer
  An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols
  ☆16Updated 4 years ago