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:
- [ICLR 2026] Official Implementation of the paper: "Two are better than one: Context window extension with multi-grained self-injection"☆12Feb 1, 2026Updated 8 months ago
- ☆17May 6, 2025Updated last year
- QLoRA for Masked Language Modeling☆23Sep 11, 2023Updated 3 years ago
- ☆23Dec 16, 2024Updated last year
- A Lisp interpreter that displays the contents of the Lisp workspace on an OLED display☆16Aug 14, 2019Updated 7 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- RESTful HTTP endpoints to RxInfer☆12Feb 23, 2026Updated 8 months ago
- Package of Pathways-on-Cloud utilities☆34Updated this week
- Untyped Concatenative Calculus☆19Dec 25, 2021Updated 4 years ago
- GET3D online data renderer☆11Dec 14, 2022Updated 3 years ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆72Apr 22, 2025Updated last year
- moondream in zig.☆76May 30, 2025Updated last year
- Code for NeurIPS 2024 paper "A SARS-CoV-2 Interaction Dataset and VHH Sequence Corpus for Antibody Language Models"☆18Oct 17, 2024Updated 2 years ago
- Creating the DeepSeek V3 model from scratch☆30Oct 8, 2026Updated 3 weeks ago
- Powerful Asynchronous Multi agent framework built from scratch in python supporting RAG and MCP compatibilities.☆12Aug 26, 2025Updated last year
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- ☆12Apr 1, 2025Updated last year
- Diffusion on the Clouds: Short-term solar energy forecasting with Diffusion Models☆48Mar 21, 2024Updated 2 years ago
- ☆13Jun 3, 2024Updated 2 years ago
- A collection of my from-scratch implementation of various models deployed on HF Spaces☆14Oct 14, 2025Updated last year
- ☆49Feb 23, 2025Updated last year
- ☆13Aug 31, 2026Updated 2 months ago
- mamba2-jax: A pure JAX/Flax implementation of Mamba-2 for language modeling and time series forecasting.☆19Jun 23, 2026Updated 4 months ago
- Implementation of the Quiet-STAR paper (https://arxiv.org/pdf/2403.09629.pdf)☆57Aug 8, 2024Updated 2 years ago
- Xtructure is datastructure for using in JAX☆23Oct 24, 2026Updated last week
- 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.
- ☆33Apr 19, 2025Updated last year
- The Expert Orchestrator AI: Dynamically Adapting, Budget-Aware, and Precisely Tailored to Your Needs☆20Jun 26, 2025Updated last year
- Crestle version of fast.ai courses☆14Nov 22, 2017Updated 8 years ago
- Fast reinforcement learning 💨☆29Jul 15, 2025Updated last year
- Train to 94% on CIFAR-10 in 4.4 seconds on a single A100☆12Dec 30, 2023Updated 2 years ago
- leveldb FUSE filesystem☆16May 15, 2014Updated 12 years ago
- PathwaysJob API is an OSS Kubernetes-native API, to deploy ML training and batch inference workloads, using Pathways on GKE.☆23Jul 9, 2026Updated 3 months ago
- A semantic search app built using only one function☆13Aug 19, 2024Updated 2 years ago
- Companion code for Grokking Megakernels: fuse an entire LLM forward pass into a single CUDA kernel☆31Feb 9, 2026Updated 8 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- AIMNet2: Fast, accurate and transferable neural network interatomic potential☆18Oct 17, 2024Updated 2 years ago
- A curated list of Radiance Fields research and resources for XR.☆11Nov 18, 2025Updated 11 months ago
- VIT inference in triton because, why not?☆36May 31, 2024Updated 2 years ago
- ☆21Mar 23, 2026Updated 7 months ago
- Simple repository for training small reasoning models☆52Aug 4, 2026Updated 2 months ago
- a multi-node fabric-attached memory manager that provides simple abstractions for accessing and allocating NVM from fabric-attached memor…☆12Jun 3, 2024Updated 2 years ago
- It's a baby compiler. (Lean btw.)☆16May 19, 2025Updated last year