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)☆50Updated 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)☆16Aug 6, 2024Updated last year
- ☆13Jun 2, 2024Updated 2 years ago
- The MDF Connect Client is the Python client to easily submit datasets to MDF Connect.☆10May 31, 2026Updated 2 weeks ago
- 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.
- Implementation of RegNetY in TensorFlow 2☆21Jan 15, 2023Updated 3 years ago
- Generative Modeling via Drifting in MLX☆43Feb 6, 2026Updated 4 months ago
- 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 last year
- 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
- A MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Dec 18, 2025Updated 5 months ago
- Parallel hyperparameter tuning with JAX☆39May 21, 2026Updated 3 weeks ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 3 years ago
- ☆107Jun 8, 2026Updated last week
- Halfedge mesh library in Rust☆29Jun 8, 2026Updated last week
- ☆15Nov 29, 2023Updated 2 years ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆65Feb 15, 2023Updated 3 years ago
- Wox plugin to search github repos, browse issues and PRs☆12Mar 28, 2021Updated 5 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
- Blazingly fast neighborhood attention☆14Nov 28, 2023Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- GBDF: Gender Balanced DeepFake Dataset☆11Jul 22, 2022Updated 3 years ago
- A Wadler–Lindig pretty printer for Python☆47Apr 20, 2026Updated last month
- Convert the TotalSegmentator dataset into the nnUNet format☆19Aug 1, 2023Updated 2 years ago
- A solver for linear complementarity problems☆12Dec 16, 2021Updated 4 years ago
- Named Tensors for Legible Deep Learning in JAX☆221Nov 8, 2025Updated 7 months ago
- CLU lets you write beautiful training loops in JAX.☆366Mar 3, 2026Updated 3 months ago
- Docker container for ocropus3 OCR system☆12Aug 19, 2018Updated 7 years ago
- A Mixed Sample Data Augmentation method for Training with Time-Frequency Domain Features☆10Oct 5, 2022Updated 3 years ago
- ANN-based Expectations Algorithm applied to the Neoclassical Investment Model☆10Mar 15, 2023Updated 3 years ago
- 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.
- ☆13Jul 12, 2024Updated last year
- Nested Sampling code☆38May 11, 2026Updated last month
- 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 last month
- Here is the repo for public scripts.☆12Jul 16, 2022Updated 3 years ago
- ☆10Dec 21, 2024Updated last year
- Python package to process videos as in Hu and Ma (2024)☆21Sep 29, 2024Updated last year