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
• google-deepmind / enn_acme
  ☆31Updated 3 years ago
• kelechi-c / dit_flow
  DiT (training + flow matching) in Jax
  ☆10Updated 9 months ago
• microsoft / Intrepid
  INTeractive learning via REPresentatIon Discovery
  ☆34Updated last year
• btnorman / First-Explore
  Repo to reproduce the First-Explore paper results
  ☆38Updated 10 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆88Updated last year
• Sea-Snell / CALM-Dialogue
  Official code for the paper "Context-Aware Language Modeling for Goal-Oriented Dialogue Systems"
  ☆34Updated 2 years ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆209Updated 2 years ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Updated last month
• sholtodouglas / scalingExperiments
  ☆62Updated 3 years ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆39Updated last month
• mgrankin / minGPT
  minGPT in JAX
  ☆48Updated 3 years ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• google-research / precondition
  ☆31Updated 4 months ago
• google-deepmind / nao_top10
  ☆19Updated 2 years ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆141Updated last year
• danielmamay / grokking
  Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.
  ☆39Updated 2 years ago
• radarFudan / mamba-minimal-jax
  ☆34Updated 11 months ago
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated 2 years ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆99Updated last month
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆43Updated 3 years ago
• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆81Updated 3 years ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 3 years ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆90Updated last year
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆53Updated 2 years ago
• Sea-Snell / Implicit-Language-Q-Learning
  Official code from the paper "Offline RL for Natural Language Generation with Implicit Language Q Learning"
  ☆209Updated 2 years ago
• danijar / ninjax
  General Modules for JAX
  ☆69Updated last month
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆38Updated last year
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆43Updated 11 months ago
• abaheti95 / LoL-RL
  Advantage Leftover Lunch Reinforcement Learning (A-LoL RL): Improving Language Models with Advantage-based Offline Policy Gradients
  ☆26Updated last year