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
- The code for experiments conducted to verify the correctness of mirror learning.☆11Jun 3, 2022Updated 4 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 3 years ago
- Graphical implementation of the Game of Life for Emacs☆17May 6, 2020Updated 6 years ago
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function☆13Apr 13, 2026Updated 2 months 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 Generalization Guarantees for (Multi-Modal) Imitation Learning☆11Jul 14, 2022Updated 3 years ago
- ☆14Jul 24, 2023Updated 2 years ago
- Tabular methods for reinforcement learning☆39Jul 3, 2020Updated 5 years ago
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆43Sep 19, 2022Updated 3 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆18Aug 8, 2022Updated 3 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
- A package for the simulation of gas chromatography (GC)☆18May 9, 2026Updated last month
- LATTICE turns retrieval into an LLM-driven navigation problem over a semantic scaffold☆37Mar 9, 2026Updated 3 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆26Jan 9, 2026Updated 5 months ago
- RND1: Scaling Diffusion Language Models☆184Feb 22, 2026Updated 3 months ago
- MacBook Pro keyboard written in SwiftUI.☆12Jan 19, 2021Updated 5 years ago
- Simple implementation of a GPT (training and inference) in PyTorch.☆13Dec 11, 2023Updated 2 years ago
- Various reinforcement learning algorithms written in Jax + Flax☆26Jun 24, 2023Updated 2 years ago
- transformer which using numpy，vision transformer of VIT, MNIST testset precision = 97.2%，mutil-attention, patch embed, position embed, fu…☆12Mar 4, 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
- A JAX Implementation of the Twin Delayed DDPG Algorithm☆35Mar 12, 2020Updated 6 years ago
- Epub reader with speech synthesis and dictionaries for Windows☆11Aug 14, 2024Updated last year
- 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.
- InternLM-7B微调, SFT/LoRA, instruction finetune☆13May 17, 2024Updated 2 years ago
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 3 years ago
- UpTeX虛擬字體與直書排版要素（12.24）.pdf☆12Mar 10, 2022Updated 4 years ago
- Materials on web programming course read in MIPT in 2015-2016☆10Dec 2, 2015Updated 10 years ago
- A C# WPF app that demonstrates uses of the NonInvasiveKeyboardHook package by letting users configure hotkeys that serve as customizable …☆11Dec 29, 2018Updated 7 years ago
- C#版屏幕截图，支持多显示器，带遮罩层☆16Jun 24, 2020Updated 5 years ago
- Ray Tracer written in Rust☆13Nov 22, 2021Updated 4 years ago
- 一个现代化的桌面聊天界面框架，专注于打造流畅优雅的对话交互体验。 优雅 对话：简约现代的聊天界面设计，支持丰富的消息类型 - 🎨 Fluent Design：采用 WPF Fluent UI，提供流畅的视觉体验 内置 OpenAI 接口，轻松对接各类 AI 服务 - 📝…☆14Jan 28, 2025Updated last year
- Jax/Flax rewrite of Karpathy's nanoGPT☆65Feb 15, 2023Updated 3 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.
- Principia is an open-source CLI tool for building robotics simulations from natural language.☆63Feb 26, 2026Updated 3 months 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☆13Jun 8, 2026Updated last week
- A simple demo of Omnivore's API☆15Jan 9, 2024Updated 2 years ago
- ☆15Jun 28, 2022Updated 3 years ago
- 粤语开放词典☆10Oct 13, 2023Updated 2 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