The simplest, fastest repository for training/finetuning medium-sized GPTs.
☆37Dec 3, 2023Updated 2 years ago
Alternatives and similar repositories for nanoGPT-jax
Users that are interested in nanoGPT-jax are comparing it to the libraries listed below
Sorting:
- Jax/Flax rewrite of Karpathy's nanoGPT☆63Feb 15, 2023Updated 3 years ago
- On the Feasibility of Cross-Task Transfer with Model-Based Reinforcement Learning☆16Apr 30, 2023Updated 2 years ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆15Jan 3, 2023Updated 3 years ago
- Actor-Sharer-Learner training framework for off-policy DRL algorithms☆22Dec 29, 2024Updated last year
- ☆13Jan 16, 2025Updated last year
- Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.☆24Sep 29, 2024Updated last year
- Platform to run interactive Reinforcement Learning agents in a Minecraft Server☆56Feb 20, 2026Updated 2 weeks ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- minGPT in JAX☆48Jan 10, 2022Updated 4 years ago
- ☆23Aug 19, 2022Updated 3 years ago
- An implementation of MuZero in JAX.☆57Nov 8, 2022Updated 3 years ago
- ☆26Jun 14, 2022Updated 3 years ago
- ☆23Jun 8, 2021Updated 4 years ago
- A MATLAB function library containing encoders, decoders and weight enumerators for Reed-Muller codes.☆11Aug 19, 2023Updated 2 years ago
- Simple JAX Graphics Library.☆36Nov 3, 2024Updated last year
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆77Jun 23, 2023Updated 2 years ago
- ☆10Apr 5, 2024Updated last year
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆93Dec 1, 2024Updated last year
- ☆13Sep 13, 2015Updated 10 years ago
- ☆12Aug 6, 2024Updated last year
- [IEEE ICASSP 2021] "A fast randomized adaptive CP decomposition for streaming tensors". In 46th IEEE International Conference on Acoustic…☆11Feb 16, 2023Updated 3 years ago
- An analog, transistor-level simulation of an 8-bit CPU in SPICE☆13Jul 29, 2021Updated 4 years ago
- Directed masked autoencoders☆14Feb 20, 2026Updated 2 weeks ago
- ☆11Sep 26, 2022Updated 3 years ago
- Julia API for Ray☆12Jan 8, 2026Updated 2 months ago
- The source code of the winning algorithm of the MICCAI Grand Challenge Carotid Artery Vessel Wall Segmentation Challenge☆14Apr 15, 2024Updated last year
- A Compiler made with python using some useful libreries.☆11Nov 11, 2019Updated 6 years ago
- ☆292Jul 15, 2024Updated last year
- LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization☆38Jan 16, 2023Updated 3 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]☆41Aug 27, 2022Updated 3 years ago
- Minimal, lightweight JAX implementations of popular models.☆205Updated this week
- Code related to the ELM neuron.☆13Feb 27, 2024Updated 2 years ago
- Matlab codes for Unsourced Multiple Access With Random User Activity by K.-H. Ngo, A. Lancho, G. Durisi, and A. Graell i Amat☆12May 26, 2023Updated 2 years ago
- ☆10Feb 20, 2024Updated 2 years ago
- Building the Bi-LSTM & the CNN-GAN models to compose Classical Music in different eras☆11Aug 2, 2021Updated 4 years ago
- Code for "DeepPolar codes", ICML 2024☆12May 7, 2024Updated last year
- ☆16Feb 28, 2026Updated last week