LLM pretraining in pure JAX
☆16Apr 1, 2026Updated 7 months ago
Alternatives and similar repositories for picodo
Users that are interested in picodo are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Minimal yet performant LLM examples in pure JAX☆286Aug 23, 2026Updated 2 months ago
- Efficient Approximate Page Rank Algorithm (based on “Local graph partitioning using PageRank vectors” by Andersen, Chung, and Lang)☆10Apr 14, 2014Updated 12 years ago
- Simple JAX Graphics Library.☆39Nov 3, 2024Updated last year
- Determining the Hierarchical Architecture of the Human Brain Using Subject-Level Clustering of Functional Networks☆14Dec 8, 2023Updated 2 years ago
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 5 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- MetaArcade is a configurable environment suite for meta-learning☆16Oct 19, 2022Updated 4 years ago
- ☆11Aug 20, 2020Updated 6 years ago
- 📄Small Batch Size Training for Language Models☆89Mar 18, 2026Updated 7 months ago
- Persistent Sampling for fast Bayesian posterior sampling and model evidence estimation in scientific applications☆14Jun 5, 2026Updated 4 months ago
- An Optax-based JAX implementation of the IVON optimizer for large-scale VI training of NNs (ICML'24 spotlight)☆10Dec 19, 2024Updated last year
- Translate Latex Formulas to R Code☆14Sep 19, 2021Updated 5 years ago
- The official repo for the CoRL 2022 paper 'Learning Control Admissibility Models with Graph Neural Networks for Multi-Agent Navigation'☆10Oct 8, 2022Updated 4 years ago
- Control Systems Analysis Framework - a framework to minimize the effort required to evaluate, implement, and verify controller design (cl…☆16Nov 21, 2022Updated 3 years ago
- Benchopt benchmark for ResNet fitting on a classification task☆12Sep 19, 2023Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Emacs control-mode client for a persistent, possibly remote tmux pane — iTerm2-style tmux integration in Emacs, rendered via Eat.☆14Updated this week
- Patches for the evince pdf viewer that make it look and function like zathura, with a status bar at the bottom and vim-like keybindings.☆11Aug 10, 2025Updated last year
- Compact LaTeX Template for the standard institute format. This is a modification of MR Bharath's LaTeX template. I've made it more compac…☆11Dec 28, 2016Updated 9 years ago
- Some of the plotting code used in our paper, as an example for good-looking plots.☆14Jan 11, 2018Updated 8 years ago
- Expected Shortfall Backtesting☆13Sep 3, 2023Updated 3 years ago
- FLoRA: A Framework for Learning Scoring Rules in Autonomous Driving Planning Systems☆13Apr 12, 2026Updated 6 months ago
- ☆51Oct 23, 2026Updated last week
- Benchmark Blackjax samplers on Inference Gym models☆15May 14, 2026Updated 5 months ago
- JAX bindings for Flash Attention v2☆110Feb 28, 2026Updated 8 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- 3D learning environment with rigid body simulation for Linux/MacOSX☆14Dec 24, 2021Updated 4 years ago
- Agentic Schema for Transparent Research Analysis.☆27Updated this week
- Calibrating LLMs with Information-Theoretic Evidential Deep Learning (ICLR 2025)☆18Mar 2, 2025Updated last year
- JAX bindings for the FlashAttention 3 kernels☆26Sep 9, 2026Updated last month
- Goal-conditioned reinforcement learning like 🔥☆15Feb 3, 2024Updated 2 years ago
- RFC-7807 compliant problem detail error response handler for FastAPI applications☆22Aug 4, 2023Updated 3 years ago
- JAX compilation of RDDL description files, and a differentiable planner in JAX.☆18Updated this week
- [CoRL 2023] Differentiable game-theoretic optimization integrated with neural networks for multi-agent prediction and control.☆16Jun 21, 2024Updated 2 years ago
- ☆26Apr 16, 2020Updated 6 years ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- qwen3 experiments☆33Jul 1, 2025Updated last year
- The COVID-19 Early Warning System (CovEWS) is a real-time early warning system for assessing individual COVID-19 related mortality risk.☆17Apr 25, 2021Updated 5 years ago
- Code exploring the use of reward machines in the context of cooperative multi-agent reinforcement learning.☆14Apr 29, 2023Updated 3 years ago
- ☆91Mar 12, 2026Updated 7 months ago
- Planning from Pixels in Environments with Combinatorially Hard Search Spaces -- NeurIPS 2021☆12Jan 7, 2022Updated 4 years ago
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆83Oct 16, 2026Updated 2 weeks ago
- R package providing a gadget to seamlessly initialise robust shiny related projects☆25Jul 9, 2021Updated 5 years ago