Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
☆47May 31, 2024Updated last year
Alternatives and similar repositories for safejax
Users that are interested in safejax are comparing it to the libraries listed below
Sorting:
- A small MNIST-like The Simpsons character database to at least have some fun while training neural networks.☆12May 12, 2021Updated 4 years ago
- 🚂 Fine-tune OpenAI models for text classification, question answering, and more☆17May 1, 2023Updated 2 years ago
- 🧠 ResNet: Deep Residual Learning for Image Recognition☆10Sep 18, 2021Updated 4 years ago
- JAX implementation of GPTQ quantization algorithm☆10Jul 19, 2023Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆47Oct 26, 2020Updated 5 years ago
- Repository for opt-out requests.☆10Mar 25, 2024Updated last year
- A Rust crate offering similar functionality to the Python transformers package using Candle.☆14Nov 19, 2024Updated last year
- Experiments in applying interpretability techniques to learned reward functions.☆10Dec 11, 2020Updated 5 years ago
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆27Feb 1, 2025Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 2 years ago
- 🦠 COVID-19 Daily Data from Worldometers with Python☆13Feb 28, 2021Updated 5 years ago
- Baselines for gymnax 🤖☆74Apr 3, 2023Updated 2 years ago
- Code to reproduce Supervised Policy Update (ICLR 2019)☆17Dec 8, 2022Updated 3 years ago
- Helpful functions for use alongside the rich Python library.☆20Aug 26, 2025Updated 6 months ago
- ☆22Nov 8, 2021Updated 4 years ago
- 🤗 Collection of examples on how to train, deploy and monitor HuggingFace models in Google Cloud Vertex AI☆22Feb 26, 2024Updated 2 years ago
- A gym environment for Stuart Armstrong's model of a treacherous turn.☆18Jul 28, 2018Updated 7 years ago
- [WIP] A 🔥 interface for running code in the cloud☆86Feb 24, 2023Updated 3 years ago
- Autoregressive transformer in JAX from scratch☆23Jan 28, 2022Updated 4 years ago
- ☆20Mar 14, 2021Updated 4 years ago
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated last year
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Jan 20, 2024Updated 2 years ago
- ☆54Apr 11, 2023Updated 2 years ago
- minGPT in JAX☆48Jan 10, 2022Updated 4 years ago
- ☆26Sep 15, 2022Updated 3 years ago
- Various reinforcement learning algorithms written in Jax + Flax☆26Jun 24, 2023Updated 2 years ago
- 🤝 Trade any tensors over the network☆31Sep 27, 2023Updated 2 years ago
- Unofficial JAX implementations of deep learning research papers☆161Jun 25, 2022Updated 3 years ago
- 🎈 A Python package template using pyproject.toml, hatch, pre-commit, black, ruff, and mkdocs.☆59Sep 7, 2023Updated 2 years ago
- Typescript wrapper for the Hugging Face Inference API.☆28Mar 2, 2023Updated 3 years ago
- Performant, differentiable reinforcement learning☆23Jun 16, 2023Updated 2 years ago
- Torch test module☆24Mar 8, 2016Updated 10 years ago
- Map-Elites based on Evolution Strategies☆33Feb 11, 2022Updated 4 years ago
- Implementation of the skill discovery algorithm described in ICLR submission "Option Discovery using Deep Skill Chaining"☆30Sep 24, 2019Updated 6 years ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆695Jan 26, 2026Updated last month
- Exponential families for JAX☆76Updated this week
- TensorFlow Examples for Natural Language Processing☆32Nov 3, 2018Updated 7 years ago
- ☆30Jun 4, 2022Updated 3 years ago
- JAX implementation ViT-VQGAN☆82Sep 21, 2022Updated 3 years ago