This repository contain the simple llama3 implementation in pure jax.
☆72Feb 17, 2025Updated last year
Alternatives and similar repositories for Llama-3-From-Scratch-In-Pure-Jax
Users that are interested in Llama-3-From-Scratch-In-Pure-Jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- QLoRA for Masked Language Modeling☆23Sep 11, 2023Updated 3 years ago
- qwen3 experiments☆33Jul 1, 2025Updated last year
- WhaleSearch 🐳 - semantic search engine. (SSE)☆16Jan 4, 2025Updated last year
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- JAX implementation of GPTQ quantization algorithm☆10Jul 19, 2023Updated 3 years 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.
- working implimention of deepseek MLA☆44Jan 8, 2025Updated last year
- mamba2-jax: A pure JAX/Flax implementation of Mamba-2 for language modeling and time series forecasting.☆19Jun 23, 2026Updated 3 months ago
- Inference code for LLaMA models in JAX☆120May 21, 2024Updated 2 years ago
- ☆13Jun 3, 2024Updated 2 years ago
- ☆33Apr 19, 2025Updated last year
- Creating the DeepSeek V3 model from scratch☆30Mar 28, 2025Updated last year
- ☆49Feb 23, 2025Updated last year
- AIMNet2: Fast, accurate and transferable neural network interatomic potential☆18Oct 17, 2024Updated last year
- It's a baby compiler. (Lean btw.)☆16May 19, 2025Updated last year
- 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.
- Crestle version of fast.ai courses☆14Nov 22, 2017Updated 8 years ago
- Xtructure is datastructure for using in JAX☆23Sep 25, 2026Updated 2 weeks ago
- ☆362Updated this week
- The Expert Orchestrator AI: Dynamically Adapting, Budget-Aware, and Precisely Tailored to Your Needs☆20Jun 26, 2025Updated last year
- VIT inference in triton because, why not?☆36May 31, 2024Updated 2 years ago
- convert a saved pytorch model to gguf and generate as much corresponding ggml c code as possible☆15Dec 19, 2023Updated 2 years ago
- Simple repository for training small reasoning models☆52Aug 4, 2026Updated 2 months ago
- Fast reinforcement learning 💨☆29Jul 15, 2025Updated last year
- NanoGPT-speedrunning for the poor T4 enjoyers☆72Apr 22, 2025Updated last year
- 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.
- ☆18Dec 2, 2024Updated last year
- ☆29Jan 14, 2025Updated last year
- Make-A-Video Latent Diffusion Model☆19Nov 15, 2023Updated 2 years ago
- ANE accelerated embedding models!☆20Dec 11, 2024Updated last year
- image captioninggg🐳☆12Aug 30, 2024Updated 2 years ago
- Full finetuning of large language models without large memory requirements☆92Sep 22, 2025Updated last year
- NeurIPS 2019: DQN(λ) = Deep Q-Network + λ-returns.☆25May 20, 2024Updated 2 years ago
- Create cohorts from databases utilizing the OMOP CDM☆10May 19, 2025Updated last year
- Diffusion on the Clouds: Short-term solar energy forecasting with Diffusion Models☆48Mar 21, 2024Updated 2 years 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.
- Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"☆92Sep 12, 2025Updated last year
- Base for the Viral Genomics and Bioinformatics Repository☆17Jul 5, 2024Updated 2 years ago
- Type inference implementation in OCaml using Algorithm W☆10Aug 26, 2021Updated 5 years ago
- Mention any three favourite things and get recommendations in the form of a flow chart by Claude Haiku.☆14Apr 6, 2024Updated 2 years ago
- ☆30Jan 26, 2023Updated 3 years ago
- ☆15Jun 5, 2023Updated 3 years ago
- ☆25May 23, 2025Updated last year