A collection of open-source large language model (LLM) implementations in JAX & Flax
☆24Apr 1, 2025Updated last year
Alternatives and similar repositories for awesome-jax-flax-llms
Users that are interested in awesome-jax-flax-llms are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Code from PLDI '21 paper "Provable Repair of Deep Neural Networks."☆10Nov 26, 2022Updated 3 years ago
- A LALR(1) OpenQASM Parser and AST Generator.☆14Jul 29, 2024Updated 2 years ago
- The implementation of an SMTLib dialect for xDSL☆19Apr 16, 2026Updated 6 months ago
- Open source platform for the privacy-preserving machine learning lifecycle☆17Aug 18, 2023Updated 3 years ago
- Classes/structs, pattern matching, safe counters... and more! Your one-stop library for programming tools not already in core Typst.☆30Oct 16, 2026Updated 2 weeks ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Indexed functors, monads and comonads that require extensions to Haskell98☆11May 18, 2024Updated 2 years ago
- ☆16Jul 4, 2026Updated 3 months ago
- Try a tactic at each step in a Lean proof.☆40Oct 13, 2026Updated 2 weeks ago
- Lazily mount a git repo without cloning it. Works with ordinary git commands☆18Jul 3, 2026Updated 3 months ago
- Google Drive filesystem adapter for just-bash☆10Apr 5, 2026Updated 6 months ago
- Analogous Safe-state Exploration (ASE) is an algorithm for provably safe and optimal exploration in MDPs with unknown, stochastic dynamic…☆10Mar 22, 2021Updated 5 years ago
- Coordinates in JAX☆38Updated this week
- A Metal-based CUDA compatibility framework for running Numba CUDA workloads on Apple Silicon.☆18Updated this week
- 동시성 프로그래밍 입문☆19Mar 13, 2022Updated 4 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.
- A enumerator for MLIR, relying on the information given by IRDL.☆26Oct 8, 2026Updated 3 weeks ago
- ☆11Sep 23, 2021Updated 5 years ago
- Lipschitz Lifelong RL☆11Nov 6, 2020Updated 5 years ago
- An open-source session replay tool for single-page applications that uses AI analysis, aggregated trends, and a RAG chatbot to help devel…☆11Jan 23, 2026Updated 9 months ago
- Jump to the local IDE source code while clicking the element of the browser automatically.☆11Updated this week
- Teradata Connector☆16Nov 17, 2025Updated 11 months ago
- Stochastic trace estimation using JAX☆22Aug 27, 2026Updated 2 months ago
- Tools for JAX☆51Oct 23, 2026Updated last week
- The repository contains code for the project "Explain the uncertain: Stochastic Shapley Values for Gaussian Process Models"☆17Jul 30, 2024Updated 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.
- Python interface for COSMO.jl convex optimisation solver.☆18Sep 27, 2021Updated 5 years ago
- Keras implementation of guide actor-critic for continuous control☆11Mar 12, 2018Updated 8 years ago
- My emacs configuration using org-mode☆13Oct 22, 2026Updated last week
- Implementing a first hurdle for expected returns☆19Updated this week
- Modular, YAML-configured Python pipeline for multi-scale fMRI functional brain network analysis. Validated on OpenNeuro ds007318 with Sch…☆15Oct 25, 2026Updated last week
- ☆14May 30, 2019Updated 7 years ago
- Implementation prototype of the Deep Deterministic Off-Policy Gradient (DD-OPG) method.☆11Jun 12, 2019Updated 7 years ago
- TensorRight: Automated Verification of Tensor Graph Rewrites☆28Oct 3, 2026Updated 3 weeks ago
- Semantic Unit Testing☆19Apr 10, 2025Updated last year
- 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.
- A shell-friendly hyperparameter search tool inspired by Optuna☆18Dec 17, 2024Updated last year
- source code of “CMS-LSTM: Context-Embedding and Multi-Scale Spatiotemporal-Expression LSTM for Video Prediction”☆10Mar 26, 2022Updated 4 years ago
- lean-related presentations☆18Oct 21, 2026Updated last week
- MLIR+EqSat☆32Jan 10, 2026Updated 9 months ago
- ☆20Oct 5, 2025Updated last year
- A practical guide to building, automating, and deploying PDFs with Typst☆18Apr 20, 2026Updated 6 months ago
- Deploy AI models at scale. High-throughput serving engine for AI/ML models that uses the latest state-of-the-art model deployment techniq…☆10May 21, 2024Updated 2 years ago