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
Sorting:
- ☆13Jun 2, 2024Updated last year
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Sep 29, 2021Updated 4 years ago
- Additional multi-backend functionality for Keras 3.☆16Mar 1, 2024Updated 2 years ago
- Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)☆17Aug 6, 2024Updated last year
- 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 8 months ago
- Convert the TotalSegmentator dataset into the nnUNet format☆19Aug 1, 2023Updated 2 years ago
- ☆32Feb 4, 2026Updated last month
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 2 years ago
- An implementation of ESM2 in Equinox+JAX☆36Jun 5, 2025Updated 9 months ago
- ☆24Jul 31, 2023Updated 2 years ago
- Parallel hyperparameter tuning with JAX☆39Jul 21, 2025Updated 7 months ago
- A text embedding extension for the Polars Dataframe library.☆29Nov 21, 2024Updated last year
- Flow-matching algorithms in JAX☆116Aug 12, 2024Updated last year
- Jax/Flax rewrite of Karpathy's nanoGPT☆63Feb 15, 2023Updated 3 years ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆58Jan 13, 2021Updated 5 years ago
- Named Tensors for Legible Deep Learning in JAX☆217Nov 8, 2025Updated 3 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆37Dec 3, 2023Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆133Feb 20, 2026Updated 2 weeks ago
- Toolkit for structure-based deep learning on RNA.☆44Feb 9, 2026Updated 3 weeks 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
- Tokamax: A GPU and TPU kernel library.☆179Updated this week
- Repository for go shared libraries (for now).☆11Dec 1, 2025Updated 3 months ago
- Applications leveraged by Statistics Canada, ready to launch on Kubernetes using Helm.☆10Dec 8, 2025Updated 2 months ago
- Econ5821 2026☆13Feb 28, 2026Updated last week
- ☆13Jan 8, 2024Updated 2 years ago
- The Easy Visual Question Answering dataset.☆34Oct 3, 2023Updated 2 years ago
- TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.☆88Sep 24, 2021Updated 4 years ago
- Topology library for Coq☆12Dec 24, 2015Updated 10 years ago
- Python package to process videos as in Hu and Ma (2024)☆20Sep 29, 2024Updated last year
- ☆11Oct 2, 2025Updated 5 months ago
- Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.☆37Aug 27, 2023Updated 2 years ago
- Computer Science, Data Science and ML Fundamentals☆11May 30, 2025Updated 9 months ago
- A microservice to encapsulate our inventory management functionality☆13Oct 25, 2023Updated 2 years ago
- An Awesome list of AI tools powered by ChatGPT / Whisper and Stable DIffusion or are useful to developers of that domain☆10Jul 26, 2023Updated 2 years ago
- QUIC pluggable crypto to use the protocol as plaintext (for use when cryptography is already handled at another layer, e.g. Wireguard)☆10Aug 27, 2025Updated 6 months ago
- TensorFlow 2 / Lite implementation of Ultra-Fast Structure-Aware Lane Detection☆12Aug 19, 2020Updated 5 years ago
- The source code of the winning algorithm of the MICCAI Grand Challenge Carotid Artery Vessel Wall Segmentation Challenge☆14Apr 15, 2024Updated last year