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:
- ☆17May 6, 2025Updated last year
- qwen3 experiments☆33Jul 1, 2025Updated last year
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- A Lisp interpreter that displays the contents of the Lisp workspace on an OLED display☆16Aug 14, 2019Updated 7 years ago
- JAX implementation of GPTQ quantization algorithm☆10Jul 19, 2023Updated 3 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.
- mamba2-jax: A pure JAX/Flax implementation of Mamba-2 for language modeling and time series forecasting.☆18Jun 23, 2026Updated last month
- Inference code for LLaMA models in JAX☆120May 21, 2024Updated 2 years ago
- ☆13Jun 3, 2024Updated 2 years ago
- Implementation of the Quiet-STAR paper (https://arxiv.org/pdf/2403.09629.pdf)☆57Aug 8, 2024Updated 2 years ago
- ☆49Feb 23, 2025Updated last year
- It's a baby compiler. (Lean btw.)☆16May 19, 2025Updated last year
- Xtructure is datastructure for using in JAX☆23Aug 2, 2026Updated 2 weeks ago
- ☆360Aug 6, 2026Updated last week
- A semantic search app built using only one function☆12Aug 19, 2024Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- High quality implementations of imitation and inverse reinforcement learning algorithms☆24Aug 19, 2025Updated 11 months ago
- VIT inference in triton because, why not?☆36May 31, 2024Updated 2 years ago
- Backup fork of Reflio to be independently maintained.☆14Apr 19, 2023Updated 3 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
- Fast reinforcement learning 💨☆29Jul 15, 2025Updated last year
- Simple repository for training small reasoning models☆51Aug 4, 2026Updated 2 weeks ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆72Apr 22, 2025Updated last year
- ☆590Jul 11, 2024Updated 2 years ago
- Run ops on Apple ANE in NPU register with pure python on M1 Asahi Linux. No Espresso, No CoreML, no metal, no .mlmodels file, no .hwx fil…☆17Jun 28, 2026Updated last month
- 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.
- In this project, we propose to study Vision Transformers trained using the Barlow Twins self-supervised method, and compare the results w…☆17Oct 3, 2023Updated 2 years ago
- Make-A-Video Latent Diffusion Model☆19Nov 15, 2023Updated 2 years ago
- A minimal, didactic implementation of tensors in Rust.☆13Feb 21, 2025Updated last year
- Implementation of a holodeck, written in Pytorch☆19Nov 1, 2023Updated 2 years ago
- Full finetuning of large language models without large memory requirements☆92Sep 22, 2025Updated 10 months ago
- ANE accelerated embedding models!☆20Dec 11, 2024Updated last year
- image captioninggg🐳☆12Aug 30, 2024Updated last year
- ☆17Dec 3, 2025Updated 8 months ago
- Simple and efficient pytorch-native transformer training and inference (batched)☆78Apr 2, 2024Updated 2 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.
- A Google Spreadsheet add-on that converts selected tables to Markdown table formats.☆12Jan 8, 2023Updated 3 years ago
- A toy nanopass compiler for x86 written in lean☆15Oct 25, 2025Updated 9 months ago
- Whalegrad 🐳 is a lightweight deep learning library written in C.☆10Jan 5, 2025Updated last year
- LLM RP TUI for Power Users.☆36Jan 13, 2026Updated 7 months ago
- An Astro app on Cloudflare Workers using the Whisper model of Workers AI☆11Feb 18, 2025Updated last year
- DACO - Decentralized autonomous charity organization☆10Dec 11, 2017Updated 8 years ago
- ☆25May 23, 2025Updated last year