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:
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆185May 7, 2025Updated last year
- A high-efficiency text embedding and reranking model based on RWKV architecture.☆22Sep 12, 2026Updated last week
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 3 years ago
- Code for the paper: Probabilistic Forecasting with Stochastic Interpolants and Follmer Processes (generative AI for forecasting)☆20May 17, 2026Updated 4 months ago
- Repository of models in Pyro☆31Jul 25, 2024Updated 2 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.
- Tabular methods for reinforcement learning☆39Jul 3, 2020Updated 6 years ago
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function☆13Apr 13, 2026Updated 5 months ago
- JAX implementation of RL algorithms and vectorized environments☆50Dec 26, 2023Updated 2 years ago
- Code for Generalization Guarantees for (Multi-Modal) Imitation Learning☆11Jul 14, 2022Updated 4 years ago
- A collection of meta-learning algorithms in Jax☆24Sep 3, 2022Updated 4 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆18Aug 8, 2022Updated 4 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 3 years ago
- make your statistical research faster☆12Jul 7, 2023Updated 3 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.
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Jan 9, 2022Updated 4 years ago
- Augmented Reality in React 16☆26Feb 2, 2019Updated 7 years ago
- A JAX Implementation of the Twin Delayed DDPG Algorithm☆35Mar 12, 2020Updated 6 years ago
- DAY R Survival Modding Guide☆17Jul 30, 2026Updated last month
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 3 years ago
- [ICML 2025] Flat-LoRA: Low-Rank Adaptation over a Flat Loss Landscape☆18May 18, 2025Updated last year
- RAD: Reinforcement Learning with Augmented Data (code for procgen experiments)☆19Mar 29, 2021Updated 5 years ago
- Ray Tracer written in Rust☆13Nov 22, 2021Updated 4 years ago
- A python module to plot secondary structure schemes☆26Mar 18, 2026Updated 6 months 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.
- JAX tutorials for PyTorch users☆14Feb 18, 2023Updated 3 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 6 years ago
- The official GitHub page for the survey paper "From Models to Systems: A Comprehensive Survey of Efficient Multimodal Learning". And this…☆21Sep 18, 2026Updated last week
- [NAACL 2024 Findings] Evaluation suite for the systematic evaluation of instruction selection methods.☆24Jul 26, 2023Updated 3 years ago
- Go Gemini - A Go SDK for Google Gemini LLM☆18Dec 17, 2023Updated 2 years ago
- A worker pool library for Rust☆13Jul 27, 2026Updated last month
- Code for our paper: Hierarchical RL Using an Ensemble of Proprioceptive Periodic Policies☆16Feb 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 3 years ago
- Materials for short course on Bayesian inference at the Data Science Summer School☆13Jun 28, 2019Updated 7 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.
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47May 31, 2024Updated 2 years ago
- ☆18Dec 30, 2023Updated 2 years ago
- a minimalistic todo app☆10May 10, 2023Updated 3 years ago
- ☆23Aug 14, 2023Updated 3 years ago
- Bookmark directories for easy directory-hopping in the terminal☆14Sep 10, 2025Updated last year
- Implementation of different variants of Parseval metric☆14Nov 3, 2014Updated 11 years ago
- Sudoku solver in Golang☆10Sep 6, 2020Updated 6 years ago