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:
- JAX implementations of various deep reinforcement learning algorithms.☆25Feb 2, 2025Updated last year
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function☆13Updated this week
- Repository of models in Pyro☆30Jul 25, 2024Updated last year
- JAX implementation of RL algorithms and vectorized environments☆51Dec 26, 2023Updated 2 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- 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
- Auxiliary variable Markov chain Monte Carlo methods☆10Oct 24, 2017Updated 8 years ago
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆25Jan 9, 2026Updated 3 months 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
- 📝 Papers I read and notes/reviews I made. Also useful links to courses (RL/NLP/Bio/QC/DevOps)☆10May 4, 2021Updated 4 years ago
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- A repository of machine-learnable formatted oklahoma o&g production data.☆10Mar 20, 2017Updated 9 years ago
- Ray Tracer written in Rust☆13Nov 22, 2021Updated 4 years ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆64Feb 15, 2023Updated 3 years ago
- Augmented Reality in React 16☆26Feb 2, 2019Updated 7 years ago
- JAX tutorials for PyTorch users☆13Feb 18, 2023Updated 3 years ago
- A worker pool library for Rust☆12Mar 23, 2026Updated 3 weeks ago
- Awesome Reinforcement Learning from Human Feedback, the secret behind ChatGPT XD☆23Dec 13, 2022Updated 3 years ago
- Code for our SIGGRAPH 2023 paper, "Acting as Inverse Inverse Planning"☆20Apr 21, 2023Updated 2 years ago
- Program that automatically locks the screen when a face is not detected☆14Jun 18, 2013Updated 12 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- 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 8 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 last month
- Fitting stochastic blockmodels to graphs☆17Jul 8, 2016Updated 9 years ago
- a minimalistic todo app☆10May 10, 2023Updated 2 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
- Align the token outputs from Spacy and Huggingface to help understand what language structures transformers see☆43May 16, 2022Updated 3 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Bookmark directories for easy directory-hopping in the terminal☆13Sep 10, 2025Updated 7 months ago
- ☆17Dec 30, 2023Updated 2 years ago
- StrategyQA 데이터 세트 번역☆22Apr 12, 2024Updated 2 years ago
- Sudoku solver in Golang☆10Sep 6, 2020Updated 5 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Apr 8, 2026Updated last week
- ☆13Mar 7, 2024Updated 2 years ago
- Markbox (FKA Bookmarky) is tag-based bookmarking tool inspired by Pinboard.in☆15Oct 9, 2022Updated 3 years ago