alvarobartt / safejaxView external linksLinks
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
- 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
- A Rust crate offering similar functionality to the Python transformers package using Candle.☆14Nov 19, 2024Updated last year
- 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 4 years ago
- Monitoring a PyTorch Lightning CNN with Weights & Biases☆15Jul 26, 2021Updated 4 years ago
- Baselines for gymnax 🤖☆74Apr 3, 2023Updated 2 years ago
- Inference code for LLaMA models in JAX☆120May 21, 2024Updated last year
- diffusionjax is a simple and accessible diffusion models package in JAX☆50Jan 19, 2025Updated last year
- Helpful functions for use alongside the rich Python library.☆20Aug 26, 2025Updated 5 months ago
- 🤗 Collection of examples on how to train, deploy and monitor HuggingFace models in Google Cloud Vertex AI☆21Feb 26, 2024Updated last year
- 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 2 years ago
- Autoregressive transformer in JAX from scratch☆23Jan 28, 2022Updated 4 years ago
- ☆21Mar 14, 2021Updated 4 years ago
- ☆91Jan 21, 2026Updated 3 weeks ago
- PIX is an image processing library in JAX, for JAX.☆433Mar 6, 2025Updated 11 months ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Jan 20, 2024Updated 2 years ago
- ☆53Apr 11, 2023Updated 2 years ago
- ☆26Sep 15, 2022Updated 3 years ago
- Various reinforcement learning algorithms written in Jax + Flax☆26Jun 24, 2023Updated 2 years ago
- Open source demo for the paper Learning to Score Behaviors for Guided Policy Optimization☆24Jun 24, 2020Updated 5 years ago
- A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. a…☆23Dec 22, 2020Updated 5 years ago
- 🎈 A Python package template using pyproject.toml, hatch, pre-commit, black, ruff, and mkdocs.☆59Sep 7, 2023Updated 2 years ago
- TensorFlow Serving + Streamlit!☆24Aug 5, 2021Updated 4 years ago
- Performant, differentiable reinforcement learning☆23Jun 16, 2023Updated 2 years ago
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆30May 19, 2022Updated 3 years ago
- Implementation of the skill discovery algorithm described in ICLR submission "Option Discovery using Deep Skill Chaining"☆30Sep 24, 2019Updated 6 years ago
- Run CellProfiler on Terra. Contains workflows that enable a full end-to-end Cell Painting pipeline.☆11May 22, 2024Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆693Jan 26, 2026Updated 3 weeks ago
- Exponential families for JAX☆76Feb 1, 2026Updated 2 weeks ago
- 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
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆40Feb 9, 2026Updated last week
- Some small scale experiments for my blog posts 📝☆80Jun 22, 2022Updated 3 years ago