LLM pretraining in pure JAX
☆16Apr 1, 2026Updated 3 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☆267Jul 4, 2026Updated 2 weeks ago
- Simple JAX Graphics Library.☆38Nov 3, 2024Updated last year
- fMRI pre-processing project using fMRIPrep, Nilearn on a HPC server.☆11Sep 30, 2022Updated 3 years ago
- MetaArcade is a configurable environment suite for meta-learning☆16Oct 19, 2022Updated 3 years ago
- Determining the Hierarchical Architecture of the Human Brain Using Subject-Level Clustering of Functional Networks☆13Dec 8, 2023Updated 2 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.
- Efficient Approximate Page Rank Algorithm (based on “Local graph partitioning using PageRank vectors” by Andersen, Chung, and Lang)☆10Apr 14, 2014Updated 12 years ago
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 4 years ago
- Reorganize fmri files to make them BIDS compliant.☆13Jul 3, 2019Updated 7 years ago
- ☆11Aug 20, 2020Updated 5 years ago
- 📄Small Batch Size Training for Language Models☆82Mar 18, 2026Updated 4 months ago
- JAX port of FLUX.1 models using flax.nnx☆23Sep 28, 2024Updated last year
- CAG : Context Awareness Gate☆15Dec 2, 2024Updated last year
- A python-numpy implementation of Ray Tracing in One Weekend☆19Aug 31, 2020Updated 5 years ago
- Translate Latex Formulas to R Code☆14Sep 19, 2021Updated 4 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.
- 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
- The official repo for the CoRL 2022 paper 'Learning Control Admissibility Models with Graph Neural Networks for Multi-Agent Navigation'☆10Oct 8, 2022Updated 3 years ago
- Benchopt benchmark for ResNet fitting on a classification task☆12Sep 19, 2023Updated 2 years ago
- 3D learning environment with rigid body simulation for Linux/MacOSX☆14Dec 24, 2021Updated 4 years ago
- JAX bindings for Flash Attention v2☆108Feb 28, 2026Updated 4 months ago
- Stochastic Gradient MCMC algorithms implemented in theano (and autograd)☆10Jan 24, 2017Updated 9 years ago
- FLoRA: A Framework for Learning Scoring Rules in Autonomous Driving Planning Systems☆13Apr 12, 2026Updated 3 months ago
- Expected Shortfall Backtesting☆13Sep 3, 2023Updated 2 years ago
- Benchmark Blackjax samplers on Inference Gym models☆15May 14, 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.
- Goal-conditioned reinforcement learning like 🔥☆15Feb 3, 2024Updated 2 years ago
- Scattering GCN: overcoming oversmoothness in graph convolutional networks☆25Jun 6, 2022Updated 4 years ago
- The repository contains code for Non-additive reward functions in Reinforcement Learning.☆15Jul 25, 2023Updated 2 years ago
- [ICRA'25] Decentralized Safe and Scalable Multi-Agent Control Under Limited Actuation☆19Jan 28, 2025Updated last year
- Add a tqdm progress bar to your JAX scans and loops.☆131May 22, 2026Updated last month
- Calibrating LLMs with Information-Theoretic Evidential Deep Learning (ICLR 2025)☆18Mar 2, 2025Updated last year
- JAX bindings for the FlashAttention 3 kernels☆25Mar 9, 2026Updated 4 months ago
- This repository contains a list of awesome literature in learning-to-rank.☆13Sep 18, 2023Updated 2 years ago
- JAX compilation of RDDL description files, and a differentiable planner in JAX.☆16Jun 28, 2026Updated 3 weeks 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.
- ☆87Mar 12, 2026Updated 4 months ago
- Fast and easy echarts with data.table backend for wrangling and a simple API☆24Jul 14, 2026Updated last week
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆56May 21, 2023Updated 3 years ago
- [CoRL 2023] Differentiable game-theoretic optimization integrated with neural networks for multi-agent prediction and control.☆16Jun 21, 2024Updated 2 years ago
- Wrapper for OpenAI Retro envs for parallel execution☆27Dec 22, 2018Updated 7 years ago
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆80Jun 18, 2026Updated last month
- Code exploring the use of reward machines in the context of cooperative multi-agent reinforcement learning.☆14Apr 29, 2023Updated 3 years ago