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 11 months ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.β18May 24, 2026Updated 2 weeks ago
- β46Updated this week
- Analyze and model weekly calendar distributions using latent componentsβ21Updated this week
- Numerical relativity surrogate waveform in Jaxβ19Aug 14, 2025Updated 9 months ago
- Proton VPN Special Offer - Get 70% off β’ AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Ambient Log Monitoringβ16Jun 26, 2011Updated 14 years ago
- Python interface for COSMO.jl convex optimisation solver.β18Sep 27, 2021Updated 4 years ago
- Bayesian Deep Ensembles via MILE: easy to use, scikit-learn compatible and fast (JAX powered)β45May 20, 2026Updated 2 weeks ago
- [KDD2025] Modeling Time-evolving Causality over Data Streams. https://dl.acm.org/doi/10.1145/3690624.3709283β16Sep 18, 2025Updated 8 months ago
- The repository contains code for the project "Explain the uncertain: Stochastic Shapley Values for Gaussian Process Models"β17Jul 30, 2024Updated last year
- Advent of Code solutions in KDB/Qβ16Apr 29, 2026Updated last month
- Territory and language names translated in different languages in .yml files. Pulled from CLDR.β26May 19, 2011Updated 15 years ago
- β20Oct 5, 2025Updated 8 months ago
- This project has a new maintainerβ33Apr 23, 2016Updated 10 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.
- Icicle Streaming Query Languageβ29Jan 20, 2026Updated 4 months ago
- Cloudflare Worker For Session Authenticationβ12Feb 4, 2023Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.β21Sep 18, 2020Updated 5 years ago
- Coordinate Axes for JAXβ64May 15, 2026Updated 3 weeks ago
- Alias-free Bessel function synthesisβ12Dec 12, 2014Updated 11 years ago
- Package of Pathways-on-Cloud utilitiesβ30Updated this week
- Fast and memory-efficient classical machine learning operatorsβ472Jun 2, 2026Updated last week
- Tensara's GPU programming problemsβ20Apr 23, 2026Updated last month
- React + MVC = Happy Developersβ49Jan 23, 2016Updated 10 years ago
- Managed Database hosting by DigitalOcean β’ AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A Github Action that overwrites your Github Pages branch with the content of current workdir, thus deploy/publish without polluting your β¦β42Sep 6, 2025Updated 9 months ago
- A simple yet powerful media player for browser built on top of video.jsβ44Dec 9, 2025Updated 6 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 4 months ago
- Lipschitz Lifelong RLβ11Nov 6, 2020Updated 5 years ago
- 8-step pipeline for evaluating PhD research ideas to top-3 finance journal quality. Works with any AI coding assistant.β130Mar 12, 2026Updated 2 months ago
- π€π§βπ A collection of tools to help you deploy, bundle HuggingFace Spaces and related assets with ease.β21Apr 15, 2026Updated last month
- A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.β125Dec 29, 2025Updated 5 months ago
- stores the Bitcoin blockchain in a MongoDB database to allow querying of block and transaction dataβ57May 31, 2013Updated 13 years ago
- A coding agent for the browserβ24Jan 8, 2026Updated 5 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer β’ AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Implementation of a git post-clone hookβ61Aug 8, 2024Updated last year
- β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
- Headless, zero-runtime video editing using MCP and FFMPEG | Pure Bash - no Python/Node runtime neededβ21Jun 5, 2025Updated last year
- SVM Classifier in Rubyβ46Jan 16, 2011Updated 15 years ago
- Package registry for Lean/Lake.β47May 30, 2026Updated last week