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)☆50May 11, 2026Updated 2 weeks 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 months ago
- An implementation of ESM2 in Equinox+JAX☆36Apr 20, 2026Updated last month
- 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.
- Heap Implementation for Python☆10Nov 4, 2016Updated 9 years ago
- This repository shows how to efficiently process variable-length sequences in TensorFlow.☆14Apr 26, 2022Updated 4 years ago
- Hierarchical Dirichlet Process (with Split-Merge Operations), originally by Chong Wang☆18Oct 12, 2013Updated 12 years ago
- ☆35May 15, 2026Updated last week
- Neural Networks for JAX☆84Sep 24, 2024Updated last year
- A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆24Jun 8, 2025Updated 11 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
- ☆14Feb 18, 2022Updated 4 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.
- ☆14Oct 31, 2025Updated 6 months ago
- A MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Dec 18, 2025Updated 5 months ago
- Parallel hyperparameter tuning with JAX☆39Updated this week
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 3 years ago
- ☆24Jul 31, 2023Updated 2 years ago
- ☆103Updated this week
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆16May 28, 2025Updated 11 months ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆65Feb 15, 2023Updated 3 years ago
- X-ray absorption re-emission scattering - Monte Carlo simulator for X-ray obscurers☆19May 13, 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.
- 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
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆38Dec 3, 2023Updated 2 years ago
- Applications leveraged by Statistics Canada, ready to launch on Kubernetes using Helm.☆10Dec 8, 2025Updated 5 months ago
- Code for the paper "Deep FTRL-ORW: An Efficient Deep Reinforcement Learning Algorithm for Solving Imperfect Information Extensive-Form Ga…☆11Dec 1, 2022Updated 3 years ago
- A generic flux/parameter population synthesis code☆14Dec 6, 2023Updated 2 years ago
- GBDF: Gender Balanced DeepFake Dataset☆11Jul 22, 2022Updated 3 years ago
- A Wadler–Lindig pretty printer for Python☆46Apr 20, 2026Updated last month
- Convert the TotalSegmentator dataset into the nnUNet format☆20Aug 1, 2023Updated 2 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- A solver for linear complementarity problems☆12Dec 16, 2021Updated 4 years ago
- CLU lets you write beautiful training loops in JAX.☆366Mar 3, 2026Updated 2 months ago
- A Mixed Sample Data Augmentation method for Training with Time-Frequency Domain Features☆10Oct 5, 2022Updated 3 years ago
- Docker container for ocropus3 OCR system☆12Aug 19, 2018Updated 7 years ago
- ☆13Jul 12, 2024Updated last year
- In-silico design pipeline for evaluating protein structure diffusion models.☆31Jun 25, 2024Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆134May 7, 2026Updated 2 weeks ago