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 10 months ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.β15Apr 8, 2026Updated 3 weeks ago
- β44Apr 16, 2026Updated last week
- Analyze and model weekly calendar distributions using latent componentsβ21Apr 20, 2026Updated last week
- Numerical relativity surrogate waveform in Jaxβ19Aug 14, 2025Updated 8 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits β’ AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- 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 3 weeks ago
- [KDD2025] Modeling Time-evolving Causality over Data Streams. https://dl.acm.org/doi/10.1145/3690624.3709283β16Sep 18, 2025Updated 7 months ago
- The repository contains code for the project "Explain the uncertain: Stochastic Shapley Values for Gaussian Process Models"β16Jul 30, 2024Updated last year
- Advent of Code solutions in KDB/Qβ16Dec 15, 2025Updated 4 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
- This project has a new maintainerβ33Apr 23, 2016Updated 10 years ago
- Managed Kubernetes at scale on DigitalOcean β’ AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Icicle Streaming Query Languageβ28Jan 20, 2026Updated 3 months ago
- 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β64Feb 26, 2026Updated 2 months ago
- Alias-free Bessel function synthesisβ12Dec 12, 2014Updated 11 years ago
- Package of Pathways-on-Cloud utilitiesβ27Updated this week
- 8-step pipeline for evaluating PhD research ideas to top-3 finance journal quality. Works with any AI coding assistant.β122Mar 12, 2026Updated last month
- Tensara's GPU programming problemsβ18Updated this week
- π€π§βπ A collection of tools to help you deploy, bundle HuggingFace Spaces and related assets with ease.β19Apr 15, 2026Updated 2 weeks 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.
- React + MVC = Happy Developersβ49Jan 23, 2016Updated 10 years ago
- 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
- Headless, zero-runtime video editing using MCP and FFMPEG | Pure Bash - no Python/Node runtime neededβ20Jun 5, 2025Updated 10 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 3 months ago
- Lipschitz Lifelong RLβ11Nov 6, 2020Updated 5 years ago
- A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.β122Dec 29, 2025Updated 4 months ago
- 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β23Jan 8, 2026Updated 3 months ago
- AI Agents on 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.
- 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
- SVM Classifier in Rubyβ46Jan 16, 2011Updated 15 years ago
- Package registry for Lean/Lake.β46Updated this week
- Claude Code Contextual Memory Virtualisation (CMV)β70Mar 16, 2026Updated last month