LLM pretraining in pure JAX
☆16Apr 1, 2026Updated 4 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☆271Jul 4, 2026Updated last month
- Implementation of SpliceAI, Illumina's deep neural network to predict variant effects on splicing, in PyTorch.☆15Mar 3, 2023Updated 3 years ago
- Simple JAX Graphics Library.☆38Nov 3, 2024Updated last year
- MetaArcade is a configurable environment suite for meta-learning☆16Oct 19, 2022Updated 3 years ago
- Captcha image generator for Zkaptcha☆14Mar 30, 2023Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆17Aug 1, 2025Updated last year
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 4 years ago
- 📄Small Batch Size Training for Language Models☆83Mar 18, 2026Updated 4 months ago
- Demos for the paper Generalized Variational Inference (Knoblauch, Jewson & Damoulas, 2019)☆20Apr 4, 2019Updated 7 years ago
- Translate Latex Formulas to R Code☆14Sep 19, 2021Updated 4 years ago
- 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
- 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 5 months 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.
- 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
- 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
- [ICRA'25] Decentralized Safe and Scalable Multi-Agent Control Under Limited Actuation☆20Jan 28, 2025Updated last year
- Add a tqdm progress bar to your JAX scans and loops.☆132May 22, 2026Updated 2 months ago
- RFC-7807 compliant problem detail error response handler for FastAPI applications☆22Aug 4, 2023Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- JAX bindings for the FlashAttention 3 kernels☆25Mar 9, 2026Updated 5 months ago
- JAX compilation of RDDL description files, and a differentiable planner in JAX.☆16Jun 28, 2026Updated last month
- ☆87Mar 12, 2026Updated 4 months ago
- Deep Exploration Networks - Diverse Deep Generative Models for DNA, RNA and Protein Sequences☆31Oct 1, 2021Updated 4 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆57May 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
- Code exploring the use of reward machines in the context of cooperative multi-agent reinforcement learning.☆14Apr 29, 2023Updated 3 years ago
- 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
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆83Updated this week
- 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.
- Implementation of Hyena Hierarchy in JAX☆10Apr 30, 2023Updated 3 years ago
- ☆591Jul 11, 2024Updated 2 years ago
- qwen3 experiments☆33Jul 1, 2025Updated last year
- The repository contains code for Non-additive reward functions in Reinforcement Learning.☆15Jul 25, 2023Updated 3 years ago
- MIMEx: Intrinsic Rewards from Masked Input Modeling [NeurIPS 2023]☆16May 17, 2023Updated 3 years ago
- R package providing a gadget to seamlessly initialise robust shiny related projects☆25Jul 9, 2021Updated 5 years ago
- A minesweeper for R!☆22May 12, 2023Updated 3 years ago