This is a port of Mistral-7B model in JAX
☆34Jul 1, 2024Updated 2 years ago
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:
- Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)☆16Aug 6, 2024Updated 2 years ago
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Sep 29, 2021Updated 5 years ago
- The MDF Connect Client is the Python client to easily submit datasets to MDF Connect.☆10Jun 22, 2026Updated 4 months ago
- Github Action to: Check version / Test / git tag / GitHub Release / Deploy to PyPI☆21Oct 3, 2025Updated last year
- Implementing scalable LLMs in pure JAX (no third-party libraries)☆56Jun 11, 2026Updated 4 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Transformer creation from scratch using Jax.☆20Dec 8, 2024Updated last year
- Heap Implementation for Python☆10Nov 4, 2016Updated 9 years ago
- JAX models for ML☆12Mar 2, 2026Updated 8 months ago
- A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆23Jun 8, 2025Updated last year
- ☆14Feb 18, 2022Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 years ago
- Reciprocal Space Attention (RSA) for Learning Long-Range Interactions☆17Nov 7, 2025Updated 11 months ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 5 years ago
- Neural Emulator Architectures in JAX.☆28Aug 25, 2026Updated 2 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆24Jul 31, 2023Updated 3 years ago
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 3 years ago
- A non-exhaustive collection of vision transformer models implemented in TensorFlow.☆10Sep 25, 2022Updated 4 years ago
- ☆16Oct 31, 2025Updated last year
- Hierarchical Dirichlet Process (with Split-Merge Operations), originally by Chong Wang☆18Oct 12, 2013Updated 13 years ago
- Universal Transformer with Memory Tokens. Pure JAX NNX. TPU-ready.☆13May 12, 2026Updated 5 months ago
- This repository contains images from 5 categories (250 for train and 50 for test for each category)☆12Apr 4, 2020Updated 6 years ago
- ☆18Feb 16, 2024Updated 2 years ago
- MIndsDB Extension for Docker Desktop☆12Feb 12, 2024Updated 2 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.
- JAX library for training sub-4B foundation models for edge☆307Aug 28, 2024Updated 2 years ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆66Feb 15, 2023Updated 3 years ago
- A typed layer on top of Polars, for statically enforcing schema types for dataframes☆17Aug 2, 2026Updated 3 months ago
- JAX bindings for Flash Attention v2☆110Feb 28, 2026Updated 8 months ago
- Neural Networks for JAX☆84Sep 24, 2024Updated 2 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
- ☆43Oct 27, 2026Updated last week
- LLM inference with HuggingFace (experimental)☆11Feb 6, 2025Updated last year
- Topology library for Coq☆12Dec 24, 2015Updated 10 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,997Oct 25, 2026Updated last week
- CLU lets you write beautiful training loops in JAX.☆371Oct 26, 2026Updated last week
- Dataset and code for "Uncertainty-Informed Deep Transfer Learning of PFAS Toxicity"☆29Apr 24, 2023Updated 3 years ago
- JAX implementation of the Mistral 7b v0.2 model☆35Jul 3, 2024Updated 2 years ago
- Differentiable N-body code (JAX)☆11May 8, 2025Updated last year
- Solve sparse linear systems in JAX using the KLU algorithm☆48Sep 18, 2026Updated last month
- jax port of snac☆11May 12, 2024Updated 2 years ago