Alternatives and similar repositories for mistral_jax

Users that are interested in mistral_jax are comparing it to the libraries listed below

• haifeng-jin / keras-benchmarks

  View on GitHub
  ☆12Jun 2, 2024Updated last year
• sayakpaul / MLPMixer-jax2tf

  View on GitHub
  This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.
  ☆15Sep 29, 2021Updated 4 years ago
• anas-rz / k3-addons

  View on GitHub
  Additional multi-backend functionality for Keras 3.
  ☆16Mar 1, 2024Updated last year
• AakashKumarNain / nanoGPTJAX

  View on GitHub
  Implementing scalable LLMs in pure JAX (no third-party libraries)
  ☆43Feb 4, 2026Updated last week
• dtunai / xLSTM-Jax

  View on GitHub
  Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)
  ☆17Aug 6, 2024Updated last year
• cgarciae / nnx

  View on GitHub
  Neural Networks for JAX
  ☆84Sep 24, 2024Updated last year
• ml-gde / jaxgarden

  View on GitHub
  A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax
  ☆22Jun 8, 2025Updated 8 months ago
• umangjpatel / kerax

  View on GitHub
  Keras-like APIs for JAX framework
  ☆50Mar 25, 2023Updated 2 years ago
• cgarciae / jax-gravity

  View on GitHub
  ☆24Jul 31, 2023Updated 2 years ago
• patrick-kidger / esm2quinox

  View on GitHub
  An implementation of ESM2 in Equinox+JAX
  ☆36Jun 5, 2025Updated 8 months ago
• TheodoreWolf / hyperoptax

  View on GitHub
  Parallel hyperparameter tuning with JAX
  ☆39Jul 21, 2025Updated 6 months ago
• lucidrains / mlp-gpt-jax

  View on GitHub
  A GPT, made only of MLPs, in Jax
  ☆59Jun 23, 2021Updated 4 years ago
• lucidrains / GAF-microbatch-pytorch

  View on GitHub
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Jan 21, 2025Updated last year
• wdoppenberg / polars-candle

  View on GitHub
  A text embedding extension for the Polars Dataframe library.
  ☆29Nov 21, 2024Updated last year
• htoyryla / DALLE-pytorch

  View on GitHub
  Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch
  ☆59Jan 13, 2021Updated 5 years ago
• marin-community / haliax

  View on GitHub
  Named Tensors for Legible Deep Learning in JAX
  ☆218Nov 8, 2025Updated 3 months ago
• cgarciae / nanoGPT-jax

  View on GitHub
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆37Dec 3, 2023Updated 2 years ago
• openxla / tokamax

  View on GitHub
  Tokamax: A GPU and TPU kernel library.
  ☆170Feb 7, 2026Updated last week
• replicate / go

  View on GitHub
  Repository for go shared libraries (for now).
  ☆11Dec 1, 2025Updated 2 months ago
• Raffi-Hagopian / AI-CAC

  View on GitHub
  AI-CAC is a deep learning model that segments and scores coronary artery calcium (CAC) on routine non-gated, non-contrast chest CT scans.
  ☆19Jan 29, 2026Updated 2 weeks ago
• Muzammal-Naseer / KA_advanced_AI

  View on GitHub
  ☆13Jan 8, 2024Updated 2 years ago
• zhentaoshi / Econ5821

  View on GitHub
  Econ5821 2026
  ☆13Feb 5, 2026Updated last week
• sayakpaul / FunMatch-Distillation

  View on GitHub
  TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.
  ☆88Sep 24, 2021Updated 4 years ago
• Danny-Dasilva / CycleTLS-Proxy

  View on GitHub
  ☆11Oct 2, 2025Updated 4 months ago
• sayakpaul / keras-xla-benchmarks

  View on GitHub
  Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
  ☆37Aug 27, 2023Updated 2 years ago
• jeromegn / quinn-plaintext

  View on GitHub
  QUIC pluggable crypto to use the protocol as plaintext (for use when cryptography is already handled at another layer, e.g. Wireguard)
  ☆11Aug 27, 2025Updated 5 months ago
• allenanhu / interactionvideo

  View on GitHub
  Python package to process videos as in Hu and Ma (2024)
  ☆18Sep 29, 2024Updated last year
• markus-k / ultrafast-lane-detection-tf2

  View on GitHub
  TensorFlow 2 / Lite implementation of Ultra-Fast Structure-Aware Lane Detection
  ☆12Aug 19, 2020Updated 5 years ago
• dschepler / coq-topology

  View on GitHub
  Topology library for Coq
  ☆12Dec 24, 2015Updated 10 years ago
• system76 / warehouse

  View on GitHub
  A microservice to encapsulate our inventory management functionality
  ☆13Oct 25, 2023Updated 2 years ago
• jax-ml / bonsai

  View on GitHub
  Minimal, lightweight JAX implementations of popular models.
  ☆191Feb 6, 2026Updated last week
• cgarciae / ciclo

  View on GitHub
  A functional training loops library for JAX
  ☆88Feb 13, 2024Updated 2 years ago
• google / CommonLoopUtils

  View on GitHub
  CLU lets you write beautiful training loops in JAX.
  ☆366Jan 21, 2026Updated 3 weeks ago
• james77777778 / keras-image-models

  View on GitHub
  A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extrac…
  ☆47Jan 1, 2026Updated last month
• patrick-kidger / wadler_lindig

  View on GitHub
  A Wadler–Lindig pretty printer for Python
  ☆44Jan 16, 2026Updated 3 weeks ago
• DavidLandup0 / deepvision

  View on GitHub
  PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/implementations - Vision Transformer (ViT), Res…
  ☆41Jul 1, 2023Updated 2 years ago
• amckay / OptStab

  View on GitHub
  Replication material for "Optimal Automatic Stabilizers"
  ☆11Aug 9, 2021Updated 4 years ago
• livebook-dev / req_bigquery

  View on GitHub
  Conveniences for querying Google BigQuery with Req
  ☆14Jan 16, 2025Updated last year
• JulienPascal / bc-MC_Operator

  View on GitHub
  This repository contains the code to generate results from the paper "Artificial Neural Networks to solve dynamic programming problems: a…
  ☆10May 24, 2024Updated last year