Autoregressive transformer in JAX from scratch
☆23Jan 28, 2022Updated 4 years ago
Alternatives and similar repositories for jax_transformer
Users that are interested in jax_transformer are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Code for Generalization Guarantees for (Multi-Modal) Imitation Learning☆11Jul 14, 2022Updated 3 years ago
- Tabular methods for reinforcement learning☆39Jul 3, 2020Updated 5 years ago
- JAX implementation of RL algorithms and vectorized environments☆51Dec 26, 2023Updated 2 years ago
- A collection of meta-learning algorithms in Jax☆24Sep 3, 2022Updated 3 years ago
- Decision Transformer JAX - Reproduction of 'Decision Transformer: Reinforcement Learning via Sequence Modeling' in JAX and Haiku☆13Aug 14, 2024Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- LATTICE turns retrieval into an LLM-driven navigation problem over a semantic scaffold☆37Mar 9, 2026Updated 2 months ago
- Snakes package for OpenAI Gym☆10Aug 20, 2019Updated 6 years ago
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆25Jan 9, 2026Updated 4 months ago
- RND1: Scaling Diffusion Language Models☆181Feb 22, 2026Updated 3 months ago
- make your statistical research faster☆12Jul 7, 2023Updated 2 years ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Jan 9, 2022Updated 4 years ago
- RAD: Reinforcement Learning with Augmented Data (code for procgen experiments)☆19Mar 29, 2021Updated 5 years ago
- [NAACL 2024 Findings] Evaluation suite for the systematic evaluation of instruction selection methods.☆24Jul 26, 2023Updated 2 years ago
- A JAX Implementation of the Twin Delayed DDPG Algorithm☆35Mar 12, 2020Updated 6 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Materials on web programming course read in MIPT in 2015-2016☆10Dec 2, 2015Updated 10 years ago
- Ray Tracer written in Rust☆13Nov 22, 2021Updated 4 years ago
- Augmented Reality in React 16☆26Feb 2, 2019Updated 7 years ago
- This repository contains an implementation of the Batch-BKB algorithm as described in the ICML 2020 paper "Near-linear time Gaussian proc…☆13Jul 14, 2020Updated 5 years ago
- A worker pool library for Rust☆12May 4, 2026Updated 3 weeks ago
- A simple demo of Omnivore's API☆15Jan 9, 2024Updated 2 years ago
- ☆15Jun 28, 2022Updated 3 years ago
- Code for our SIGGRAPH 2023 paper, "Acting as Inverse Inverse Planning"☆20Apr 21, 2023Updated 3 years ago
- Program that automatically locks the screen when a face is not detected☆14Jun 18, 2013Updated 12 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Code for our paper: Hierarchical RL Using an Ensemble of Proprioceptive Periodic Policies☆15Feb 21, 2019Updated 7 years ago
- A Python library that tees the standard output & standard error from the current process to files on disk, while preserving terminal sema…☆16Aug 17, 2023Updated 2 years ago
- Chainer implementation of Improved Techniques for Training GANs☆10Apr 20, 2017Updated 9 years ago
- time-bomb.nvim is a minimal Neovim plugin for timers and Pomodoro cycles to boost developer focus. Features floating timers, 9 progress b…☆32Mar 12, 2026Updated 2 months ago
- Materials for short course on Bayesian inference at the Data Science Summer School☆13Jun 28, 2019Updated 6 years ago
- Fitting stochastic blockmodels to graphs☆17Jul 8, 2016Updated 9 years ago
- a minimalistic todo app☆10May 10, 2023Updated 3 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47May 31, 2024Updated last year
- Implementations of different reinforcement learning algorithms☆10Aug 23, 2018Updated 7 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- ☆18Dec 30, 2023Updated 2 years ago
- StrategyQA 데이터 세트 번역☆22Apr 12, 2024Updated 2 years ago
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆32Oct 26, 2022Updated 3 years ago
- ☆15Jan 21, 2023Updated 3 years ago
- Reproducing the paper "The continuous Bernoulli: fixing a pervasive error in variational autoencoders" for the Reproducibility Challenge …☆12Jul 25, 2024Updated last year
- Command line tool for converting images to ASCII art☆20May 1, 2026Updated 3 weeks ago
- Rust powered background process CLI task manager☆14May 2, 2022Updated 4 years ago