This is a port of Mistral-7B model in JAX
☆34Jul 1, 2024Updated 2 years ago
Alternatives and similar repositories for mistral_jax
Users that are interested in mistral_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:
- Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)☆16Aug 6, 2024Updated 2 years ago
- This repository shows how to efficiently process variable-length sequences in TensorFlow.☆14Apr 26, 2022Updated 4 years ago
- A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆23Jun 8, 2025Updated last year
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 3 years ago
- ☆24Jul 31, 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.
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- An implementation of ESM2 in Equinox+JAX☆36Apr 20, 2026Updated 3 months ago
- Neural Networks for JAX☆84Sep 24, 2024Updated last year
- X-ray absorption re-emission scattering - Monte Carlo simulator for X-ray obscurers☆21May 13, 2025Updated last year
- Jax/Flax rewrite of Karpathy's nanoGPT☆66Feb 15, 2023Updated 3 years ago
- Topology library for Coq☆12Dec 24, 2015Updated 10 years ago
- Implementing scalable LLMs in pure JAX (no third-party libraries)☆55Jun 11, 2026Updated 2 months ago
- ☆13Jun 2, 2024Updated 2 years ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Jan 21, 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.
- ☆40Aug 7, 2026Updated last week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆38Dec 3, 2023Updated 2 years ago
- A Wadler–Lindig pretty printer for Python☆48Apr 20, 2026Updated 3 months ago
- CLU lets you write beautiful training loops in JAX.☆370Jul 7, 2026Updated last month
- Getting Started in Imitation Learning☆13Mar 3, 2025Updated last year
- Named Tensors for Legible Deep Learning in JAX☆228Nov 8, 2025Updated 9 months ago
- GBDF: Gender Balanced DeepFake Dataset☆11Jul 22, 2022Updated 4 years ago
- ☆13Jul 12, 2024Updated 2 years ago
- A Mixed Sample Data Augmentation method for Training with Time-Frequency Domain Features☆10Oct 5, 2022Updated 3 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.
- JAX implementation of the Mistral 7b v0.2 model☆35Jul 3, 2024Updated 2 years ago
- Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.☆12Jun 9, 2023Updated 3 years ago
- Corpus and code for Aligned Recipe Actions (ARA) corpus, EMNLP 2021☆10May 22, 2024Updated 2 years ago
- This is the numerical approach proposed in the paper "Optimal Incentives to Mitigate Epidemics: A Stackelberg Mean Field Game Approach" b…☆13Nov 22, 2021Updated 4 years ago
- ANN-based Expectations Algorithm applied to the Neoclassical Investment Model☆10Mar 15, 2023Updated 3 years ago
- ☆10Dec 21, 2024Updated last year
- Here is the repo for public scripts.☆12Jul 16, 2022Updated 4 years ago
- Vehicle control☆11Jun 8, 2019Updated 7 years ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆142Dec 20, 2023Updated 2 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.
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- Replication material for "Optimal Automatic Stabilizers"☆11Aug 9, 2021Updated 5 years ago
- Clean up the public namespace of your package!☆59May 26, 2025Updated last year
- ☆10Jan 25, 2018Updated 8 years ago
- Research Paper - Automated Labelling of Coronary Vessels via Neural Network☆10Oct 12, 2020Updated 5 years ago
- ☆13Jan 22, 2021Updated 5 years ago
- This repository contains the code to generate results from the paper "Artificial Neural Networks to solve dynamic programming problems: a…☆10May 24, 2024Updated 2 years ago