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
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• facebookresearch / cascade
  Implementation of CASCADE in Learning General World Models in a Handful of Reward-Free Deployments (NeurIPS 22).
  ☆29Updated 2 years ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• Sea-Snell / CALM-Dialogue
  Official code for the paper "Context-Aware Language Modeling for Goal-Oriented Dialogue Systems"
  ☆34Updated 2 years ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 11 months ago
• btnorman / First-Explore
  Repo to reproduce the First-Explore paper results
  ☆37Updated 4 months ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆20Updated 10 months ago
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated last year
• srush / drop7
  ☆18Updated last year
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆10Updated last year
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 3 months ago
• google-deepmind / agent_debugger
  Causal Analysis of Agent Behavior for AI Safety
  ☆18Updated last year
• CarperAI / treasure_trove
  ☆22Updated last year
• scottlogic-alex / prm800k-denorm
  Script for processing OpenAI's PRM800K process supervision dataset into an Alpaca-style instruction-response format
  ☆27Updated last year
• mansimov / chatgpt_cli
  Lightweight wrapper of the official ChatGPT API in your terminal
  ☆43Updated 2 years ago
• lucidrains / holodeck-pytorch
  Implementation of a holodeck, written in Pytorch
  ☆18Updated last year
• lucidrains / token-shift-gpt
  Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing
  ☆50Updated 3 years ago
• subho406 / agalite
  AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)
  ☆18Updated 7 months ago
• google-deepmind / zipfian_environments
  ☆28Updated 2 years ago
• danijar / ninjax
  General Modules for JAX
  ☆66Updated last month
• brianfitzgerald / jax-mmdit
  Implementation of Diffusion Transformers and Rectified Flow in Jax
  ☆23Updated 10 months ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆32Updated 6 months ago
• prajjwal1 / rl_paradigm
  ☆17Updated last year
• Z-T-WANG / LaProp-Optimizer
  Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"
  ☆28Updated 4 years ago
• FLAIROx / cultural-accumulation
  ☆13Updated 10 months ago
• google-research / precondition
  ☆31Updated last month
• google-deepmind / enn_acme
  ☆31Updated 2 years ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆33Updated 6 months ago