Alternatives and similar repositories for t5-jax:

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

• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆20Updated 9 months ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 9 months ago
• microsoft / Intrepid
  INTeractive learning via REPresentatIon Discovery
  ☆33Updated 10 months ago
• andylolu2 / jax-vqvae-gpt
  Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.
  ☆11Updated last year
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated last year
• jax-ml / jax-llm-examples
  ☆87Updated 2 weeks ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆41Updated 5 months ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 2 months 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 / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 5 months ago
• prajjwal1 / rl_paradigm
  ☆16Updated last year
• radarFudan / mamba-minimal-jax
  ☆30Updated 4 months ago
• facebookresearch / cascade
  Implementation of CASCADE in Learning General World Models in a Handful of Reward-Free Deployments (NeurIPS 22).
  ☆29Updated 2 years ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆30Updated 5 months ago
• google-deepmind / enn_acme
  ☆31Updated 2 years ago
• ThomasMiconi / Meta-Task-Generator
  Automatically generate simple meta-learning tasks from a very large space
  ☆15Updated last year
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated 8 months ago
• andrew-silva / clean-rl-mlx
  Clean RL implementation using MLX
  ☆28Updated last year
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated 8 months ago
• chongyi-zheng / td_infonce
  ☆27Updated last year
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆203Updated last year
• FLAIROx / cultural-accumulation
  ☆13Updated 8 months ago
• btnorman / First-Explore
  Repo to reproduce the First-Explore paper results
  ☆37Updated 3 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated last year
• google-deepmind / csuite
  ☆43Updated 6 months ago
• google-deepmind / tell_me_why_explanations_rl
  ☆35Updated last year
• google-deepmind / zipfian_environments
  ☆28Updated 2 years ago
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆9Updated last year
• google-deepmind / agent_debugger
  Causal Analysis of Agent Behavior for AI Safety
  ☆17Updated last year