ztjhz / t5-jaxLinks
JAX implementation of the T5 model: Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer
☆23Updated 2 years ago
Alternatives and similar repositories for t5-jax
Users that are interested in t5-jax are comparing it to the libraries listed below
Sorting:
- Official code for the paper "Context-Aware Language Modeling for Goal-Oriented Dialogue Systems"☆34Updated 2 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆84Updated last year
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Updated last year
- Repo to reproduce the First-Explore paper results☆37Updated 6 months ago
- Machine Learning eXperiment Utilities☆46Updated 3 weeks ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks☆33Updated 8 months ago
- RL algorithm: Advantage induced policy alignment☆65Updated last year
- Lightweight wrapper of the official ChatGPT API in your terminal☆43Updated 2 years ago
- Advantage Leftover Lunch Reinforcement Learning (A-LoL RL): Improving Language Models with Advantage-based Offline Policy Gradients☆26Updated 10 months ago
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆20Updated last year
- ☆61Updated 3 years ago
- ☆22Updated last year
- Script for processing OpenAI's PRM800K process supervision dataset into an Alpaca-style instruction-response format☆27Updated 2 years ago
- ☆31Updated 7 months ago
- Clean RL implementation using MLX☆32Updated last year
- Scaling scaling laws with board games.☆49Updated 2 years ago
- INTeractive learning via REPresentatIon Discovery☆34Updated last year
- ☆13Updated last year
- ☆17Updated last year
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆71Updated 2 weeks ago
- Generative cellular automaton-like learning environments for RL.☆19Updated 5 months ago
- Causal Analysis of Agent Behavior for AI Safety☆18Updated 2 years ago
- Implementation of Direct Preference Optimization☆16Updated 2 years ago
- A library for squeakily cleaning and filtering language datasets.☆47Updated 2 years ago
- [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)☆20Updated 11 months ago
- Play chess against large language models.☆47Updated last year
- ☆31Updated last month
- ☆61Updated last year
- ☆31Updated 2 years ago