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)☆49Apr 23, 2026Updated last 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
- Heap Implementation for Python☆10Nov 4, 2016Updated 9 years ago
- 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.
- This repository shows how to efficiently process variable-length sequences in TensorFlow.☆14Apr 26, 2022Updated 4 years ago
- Implementation of RegNetY in TensorFlow 2☆21Jan 15, 2023Updated 3 years ago
- Generative Modeling via Drifting in MLX☆42Feb 6, 2026Updated 3 months ago
- Hierarchical Dirichlet Process (with Split-Merge Operations), originally by Chong Wang☆18Oct 12, 2013Updated 12 years ago
- ☆55Apr 22, 2026Updated 2 weeks ago
- ☆35Apr 27, 2026Updated last week
- Neural Networks for JAX☆84Sep 24, 2024Updated last year
- Additional multi-backend functionality for Keras 3.☆16Mar 1, 2024Updated 2 years ago
- ☆14Oct 31, 2025Updated 6 months 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 MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Dec 18, 2025Updated 4 months ago
- Parallel hyperparameter tuning with JAX☆39Apr 16, 2026Updated 2 weeks ago
- Flow-matching algorithms in JAX☆116Aug 12, 2024Updated last year
- Jax/Flax rewrite of Karpathy's nanoGPT☆64Feb 15, 2023Updated 3 years ago
- Wox plugin to search github repos, browse issues and PRs☆12Mar 28, 2021Updated 5 years ago
- X-ray absorption re-emission scattering - Monte Carlo simulator for X-ray obscurers☆19May 13, 2025Updated 11 months 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
- Blazingly fast neighborhood attention☆14Nov 28, 2023Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Applications leveraged by Statistics Canada, ready to launch on Kubernetes using Helm.☆10Dec 8, 2025Updated 4 months 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 2 weeks ago
- Convert the TotalSegmentator dataset into the nnUNet format☆20Aug 1, 2023Updated 2 years ago
- Named Tensors for Legible Deep Learning in JAX☆219Nov 8, 2025Updated 5 months ago
- CLU lets you write beautiful training loops in JAX.☆368Mar 3, 2026Updated 2 months ago
- ☆13Jul 12, 2024Updated last year
- In-silico design pipeline for evaluating protein structure diffusion models.☆31Jun 25, 2024Updated last year
- 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 JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆135Apr 17, 2026Updated 2 weeks 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.☆11Nov 23, 2023Updated 2 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
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- Clean up the public namespace of your package!☆59May 26, 2025Updated 11 months ago