A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax
☆23Jun 8, 2025Updated last year
Alternatives and similar repositories for jaxgarden
Users that are interested in jaxgarden are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Universal Transformer with Memory Tokens. Pure JAX NNX. TPU-ready.☆13May 12, 2026Updated 5 months ago
- ☆33Updated this week
- Handy tool that facilitates turning images of math equations and charts into LaTeX code, quickly and easily!☆11Sep 1, 2026Updated 2 months ago
- Implemented YOLOv2 with Tensorflow 2.0☆10Oct 6, 2022Updated 4 years ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆41Jul 20, 2026Updated 3 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.
- ADK Hackathon Fashion Project☆22Sep 10, 2025Updated last year
- Parameter-efficient transformer embeddings replace learned embeddings with hardware-aware polynomial expansions of token IDs.☆13Feb 6, 2026Updated 9 months ago
- Minimal copilot for your terminal shell, supports 8+ languages, OpenAI + Bring your own model☆18Sep 12, 2026Updated last month
- Lion - EvoLved Sign Momentum w/ New Optimizer API in TensorFlow 2.11+☆10Feb 16, 2023Updated 3 years ago
- 🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.☆17Jun 5, 2025Updated last year
- A Library That Uses Quantized Diffusion Model With Clustered Weights For Efficiently Generating More Image Datasets On-Device.☆14Jun 9, 2023Updated 3 years ago
- ☆19Sep 20, 2022Updated 4 years ago
- MeCab model trained with OpenKorPos.☆23Jun 19, 2022Updated 4 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆58Jun 26, 2022Updated 4 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.
- Lightning-like training API for JAX with Flax☆45Dec 8, 2024Updated last year
- JAX library for training sub-4B foundation models for edge☆307Aug 28, 2024Updated 2 years ago
- ☆24Jul 31, 2023Updated 3 years ago
- Google TPU optimizations for transformers models☆137Jan 23, 2026Updated 9 months ago
- A metrics library for the JAX ecosystem☆40Mar 16, 2023Updated 3 years ago
- generative language model training on top of the JAX and Huggingface 🤗☆10May 11, 2022Updated 4 years ago
- This repository holds files and scripts for incorporating simple CI/CD practices for model training in ML.☆20Oct 26, 2021Updated 5 years ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆66Feb 15, 2023Updated 3 years ago
- ☆315Oct 13, 2026Updated 3 weeks 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.
- Minimal, lightweight JAX implementations of popular models.☆246May 29, 2026Updated 5 months ago
- Tensorflow implementation of "Tackling the Generative Learning Trilemma with Denoising Diffusion GANs" (ICLR 2022 Spotlight)☆21Aug 3, 2022Updated 4 years ago
- Topology library for Coq☆12Dec 24, 2015Updated 10 years ago
- MIndsDB Extension for Docker Desktop☆12Feb 12, 2024Updated 2 years ago
- Generative Models for Simulation-Based Inference in JAX☆21Oct 27, 2026Updated last week
- Your favourite classical machine learning algos on the GPU/TPU☆23Dec 14, 2025Updated 10 months ago
- ☆15Oct 24, 2023Updated 3 years ago
- ☆13Jul 12, 2024Updated 2 years ago
- Open-source code for ''Individual Fairness for Graph Neural Networks: A Ranking based Approach''.☆11Jul 8, 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.
- Vehicle control☆12Jun 8, 2019Updated 7 years ago
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- [ICLR 2023] Eva: Practical Second-order Optimization with Kronecker-vectorized Approximation☆12Jul 31, 2023Updated 3 years ago
- This repository is created as part of Sebastian's Raschka's workshop- Building LLMs Ground Up.☆16Sep 14, 2024Updated 2 years ago
- A Golang replacement for the Kubeflow Jupyter Web APIs / Un remplacement Golang pour les API de Web de Jupyter, partie de Kubeflow☆16Updated this week
- Valentine's Day Anonymous matching☆10Jul 25, 2014Updated 12 years ago
- Search-Based Software Engineering Course☆14Oct 25, 2022Updated 4 years ago