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:
- π₯ FireChain Backend β CLI Driven β’ Real-time RTDB β’ BullMQ Workersβ13Jun 10, 2025Updated 9 months ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.β15Apr 1, 2026Updated last week
- β42Apr 1, 2026Updated last week
- Analyze and model weekly calendar distributions using latent componentsβ21Mar 30, 2026Updated last week
- Numerical relativity surrogate waveform in Jaxβ19Aug 14, 2025Updated 7 months ago
- DigitalOcean Gradient AI Platform β’ AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Ambient Log Monitoringβ16Jun 26, 2011Updated 14 years ago
- Python interface for COSMO.jl convex optimisation solver.β17Sep 27, 2021Updated 4 years ago
- Bayesian Deep Ensembles via MILE: easy to use, scikit-learn compatible and fast (JAX powered)β42Apr 1, 2026Updated last week
- The repository contains code for the project "Explain the uncertain: Stochastic Shapley Values for Gaussian Process Models"β16Jul 30, 2024Updated last year
- [KDD2025] Modeling Time-evolving Causality over Data Streams. https://dl.acm.org/doi/10.1145/3690624.3709283β16Sep 18, 2025Updated 6 months ago
- Advent of Code solutions in KDB/Qβ16Dec 15, 2025Updated 3 months ago
- Territory and language names translated in different languages in .yml files. Pulled from CLDR.β26May 19, 2011Updated 14 years ago
- β20Oct 5, 2025Updated 6 months ago
- 8-step pipeline for evaluating PhD research ideas to top-3 finance journal quality. Works with any AI coding assistant.β101Mar 12, 2026Updated 3 weeks ago
- DigitalOcean Gradient AI Platform β’ AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- This project has a new maintainerβ33Apr 23, 2016Updated 9 years ago
- Icicle Streaming Query Languageβ28Jan 20, 2026Updated 2 months ago
- Package of Pathways-on-Cloud utilitiesβ26Updated this week
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.β21Sep 18, 2020Updated 5 years ago
- Cloudflare Worker For Session Authenticationβ12Feb 4, 2023Updated 3 years ago
- Coordinate Axes for JAXβ63Feb 26, 2026Updated last month
- Alias-free Bessel function synthesisβ12Dec 12, 2014Updated 11 years ago
- Headless, zero-runtime video editing using MCP and FFMPEG | Pure Bash - no Python/Node runtime neededβ20Jun 5, 2025Updated 10 months ago
- React + MVC = Happy Developersβ49Jan 23, 2016Updated 10 years ago
- 1-Click AI Models by DigitalOcean Gradient β’ AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- A Github Action that overwrites your Github Pages branch with the content of current workdir, thus deploy/publish without polluting your β¦β42Sep 6, 2025Updated 7 months ago
- A simple yet powerful media player for browser built on top of video.jsβ44Dec 9, 2025Updated 4 months 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 2 months ago
- A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.β122Dec 29, 2025Updated 3 months ago
- Lipschitz Lifelong RLβ11Nov 6, 2020Updated 5 years ago
- A Typst editor and renderer in Obsidianβ60Mar 6, 2026Updated last month
- stores the Bitcoin blockchain in a MongoDB database to allow querying of block and transaction dataβ57May 31, 2013Updated 12 years ago
- A coding agent for the browserβ21Jan 8, 2026Updated 3 months ago
- Implementation of a git post-clone hookβ60Aug 8, 2024Updated last year
- Managed Database hosting by DigitalOcean β’ AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- β46Mar 3, 2023Updated 3 years ago
- β29Jan 26, 2024Updated 2 years ago
- Aquarium: A Comprehensive Framework for Exploring Predator-Prey Dynamics through Multi-Agent Reinforcement Learning Algorithmsβ13Apr 3, 2024Updated 2 years ago
- Package registry for Lean/Lake.β44Mar 30, 2026Updated last week
- SVM Classifier in Rubyβ46Jan 16, 2011Updated 15 years ago
- Helping sift through user concernsβ29May 2, 2025Updated 11 months ago
- Claude Code Contextual Memory Virtualisation (CMV)β66Mar 16, 2026Updated 3 weeks ago