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 last year
• Sea-Snell / CALM-Dialogue
  Official code for the paper "Context-Aware Language Modeling for Goal-Oriented Dialogue Systems"
  ☆34Updated 2 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆35Updated 10 months ago
• sholtodouglas / scalingExperiments
  ☆61Updated 3 years ago
• google-research / precondition
  ☆31Updated 2 months ago
• google-deepmind / enn_acme
  ☆31Updated 3 years ago
• microsoft / Intrepid
  INTeractive learning via REPresentatIon Discovery
  ☆34Updated last year
• btnorman / First-Explore
  Repo to reproduce the First-Explore paper results
  ☆38Updated 8 months ago
• google-deepmind / nao_top10
  ☆19Updated 2 years ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated last month
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆88Updated last year
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated last year
• augustwester / transformer-xl
  A lightweight PyTorch implementation of the Transformer-XL architecture proposed by Dai et al. (2019)
  ☆37Updated 2 years ago
• facebookresearch / cascade
  Implementation of CASCADE in Learning General World Models in a Handful of Reward-Free Deployments (NeurIPS 22).
  ☆28Updated 2 years ago
• google-research / jestimator
  Amos optimizer with JEstimator lib.
  ☆82Updated last year
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆208Updated last year
• microsoft / greenlands
  Platform to run interactive Reinforcement Learning agents in a Minecraft Server
  ☆53Updated last year
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆99Updated last week
• FLAIROx / cultural-accumulation
  ☆13Updated last year
• vwxyzjn / a2c_is_a_special_case_of_ppo
  A2C is a special case of PPO!
  ☆22Updated 3 years ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆142Updated last year
• google-deepmind / zipfian_environments
  ☆28Updated 3 years ago
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆96Updated 4 years ago
• chongyi-zheng / td_infonce
  Implementations of Temporal Difference InfoNCE (TD InfoNCE)
  ☆30Updated last year
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆33Updated 10 months ago
• google-deepmind / csuite
  ☆44Updated 11 months ago
• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆80Updated 3 years ago
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆11Updated 2 years ago
• machelreid / can-wikipedia-help-offline-rl
  Official code for "Can Wikipedia Help Offline Reinforcement Learning?" by Machel Reid, Yutaro Yamada and Shixiang Shane Gu
  ☆105Updated 3 years ago