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
- QLoRA for Masked Language Modeling☆23Sep 11, 2023Updated 2 years ago
- qwen3 experiments☆33Jul 1, 2025Updated last year
- Fixed-point scalar and matrix multiplication library for SectorLISP☆15Jan 23, 2022Updated 4 years ago
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- A Lisp interpreter that displays the contents of the Lisp workspace on an OLED display☆16Aug 14, 2019Updated 6 years ago
- Untyped Concatenative Calculus☆19Dec 25, 2021Updated 4 years ago
- working implimention of deepseek MLA☆44Jan 8, 2025Updated last year
- Code for NeurIPS 2024 paper "A SARS-CoV-2 Interaction Dataset and VHH Sequence Corpus for Antibody Language Models"☆16Oct 17, 2024Updated last year
- A tool for web browsers that simulates execution of the dynamic GoI abstract machine for the lambda-calculus.☆15Oct 1, 2021Updated 4 years ago
- Inference code for LLaMA models in JAX☆120May 21, 2024Updated 2 years ago
- Curated materials for different machine learning related summer schools☆19Mar 8, 2021Updated 5 years ago
- ☆13Jun 3, 2024Updated 2 years ago
- Creating the DeepSeek V3 model from scratch☆28Mar 28, 2025Updated last year
- 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.
- ☆49Feb 23, 2025Updated last year
- AIMNet2: Fast, accurate and transferable neural network interatomic potential☆19Oct 17, 2024Updated last year
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆22Oct 12, 2023Updated 2 years ago
- It's a baby compiler. (Lean btw.)☆16May 19, 2025Updated last year
- Rust bindings for the Voikko library☆16Mar 26, 2022Updated 4 years ago
- Xtructure is datastructure for using in JAX☆23Updated this week
- ☆13Jun 16, 2026Updated last month
- ☆358Jul 20, 2026Updated last week
- High quality implementations of imitation and inverse reinforcement learning algorithms☆24Aug 19, 2025Updated 11 months 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.
- VIT inference in triton because, why not?☆36May 31, 2024Updated 2 years ago
- Fast reinforcement learning 💨☆29Jul 15, 2025Updated last year
- Simple repository for training small reasoning models☆51Feb 17, 2026Updated 5 months ago
- Train to 94% on CIFAR-10 in 4.4 seconds on a single A100☆12Dec 30, 2023Updated 2 years ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆72Apr 22, 2025Updated last year
- ☆18Dec 2, 2024Updated last year
- ☆591Jul 11, 2024Updated 2 years ago
- GCRL in JAX. Official repository for LEO (ICML 2026).☆28Jun 20, 2026Updated last month
- Methods for specifying, checking, typing, rendering, executing, analyzing, and visualizing state space models, for Active Inference and B…☆30Jul 20, 2026Updated last week
- 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.
- 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
- NeurIPS 2019: DQN(λ) = Deep Q-Network + λ-returns.☆25May 20, 2024Updated 2 years ago
- Simple and efficient pytorch-native transformer training and inference (batched)☆78Apr 2, 2024Updated 2 years ago
- Sparse Fourier Backpropagation in Cryo-EM Reconstruction☆14Dec 3, 2023Updated 2 years ago
- ☆21Jun 26, 2023Updated 3 years ago
- Whalegrad 🐳 is a lightweight deep learning library written in C.☆10Jan 5, 2025Updated last year