This is a port of Mistral-7B model in JAX
☆33Jul 1, 2024Updated last year
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:
- Implementing scalable LLMs in pure JAX (no third-party libraries)☆48Updated this week
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Sep 29, 2021Updated 4 years ago
- Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)☆17Aug 6, 2024Updated last year
- ☆13Jun 2, 2024Updated last year
- The MDF Connect Client is the Python client to easily submit datasets to MDF Connect.☆10Mar 9, 2026Updated 2 weeks ago
- 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.
- ☆33Feb 4, 2026Updated last month
- Implementation of RegNetY in TensorFlow 2☆21Jan 15, 2023Updated 3 years ago
- Generative Modeling via Drifting in MLX☆42Feb 6, 2026Updated last month
- Neural Networks for JAX☆84Sep 24, 2024Updated last year
- A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆23Jun 8, 2025Updated 9 months ago
- An extensible library of tools that extract metadata from scientific files☆15Jan 9, 2023Updated 3 years ago
- Additional multi-backend functionality for Keras 3.☆16Mar 1, 2024Updated 2 years ago
- A MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Dec 18, 2025Updated 3 months ago
- Parallel hyperparameter tuning with JAX☆39Updated this week
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 3 years ago
- ☆24Jul 31, 2023Updated 2 years ago
- Flow-matching algorithms in JAX☆117Aug 12, 2024Updated last year
- ☆18Aug 27, 2020Updated 5 years ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆64Feb 15, 2023Updated 3 years ago
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆14May 28, 2025Updated 9 months ago
- Wox plugin to search github repos, browse issues and PRs☆12Mar 28, 2021Updated 4 years ago
- Topology library for Coq☆12Dec 24, 2015Updated 10 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
- 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.
- Applications leveraged by Statistics Canada, ready to launch on Kubernetes using Helm.☆10Dec 8, 2025Updated 3 months ago
- GBDF: Gender Balanced DeepFake Dataset☆11Jul 22, 2022Updated 3 years ago
- Convert the TotalSegmentator dataset into the nnUNet format☆19Aug 1, 2023Updated 2 years ago
- A Wadler–Lindig pretty printer for Python☆46Jan 16, 2026Updated 2 months ago
- Named Tensors for Legible Deep Learning in JAX☆217Nov 8, 2025Updated 4 months ago
- CLU lets you write beautiful training loops in JAX.☆367Mar 3, 2026Updated 3 weeks ago
- A Mixed Sample Data Augmentation method for Training with Time-Frequency Domain Features☆10Oct 5, 2022Updated 3 years ago
- ☆13Jul 12, 2024Updated last year
- Nested Sampling code☆35Jun 12, 2025Updated 9 months ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Active learning workflow developed as a part of the upcoming article "Machine Learning Inter-Atomic Potentials Generation Driven by Activ…☆28Oct 14, 2021Updated 4 years ago
- ☆13Jan 8, 2024Updated 2 years ago
- JAX implementation of the Mistral 7b v0.2 model☆35Jul 3, 2024Updated last year
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Nov 23, 2023Updated 2 years ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆141Dec 20, 2023Updated 2 years ago
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- Clean up the public namespace of your package!☆59May 26, 2025Updated 10 months ago