Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
☆47May 31, 2024Updated 2 years ago
Alternatives and similar repositories for safejax
Users that are interested in safejax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Adaptable Agent Populations via a Generative Model of Policies☆12Oct 14, 2021Updated 5 years ago
- JAX implementation of LLaMA, aiming to train LLaMA on Google Cloud TPU☆14Jul 22, 2023Updated 3 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Jan 16, 2023Updated 3 years ago
- 🍱 fsspec interface for Weights & Biases (wandb)☆10Oct 27, 2022Updated 4 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆52Jan 19, 2025Updated last year
- 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.
- Experiments in applying interpretability techniques to learned reward functions.☆10Dec 11, 2020Updated 5 years ago
- 🧠 ResNet: Deep Residual Learning for Image Recognition☆10Sep 18, 2021Updated 5 years ago
- ☆18Aug 24, 2024Updated 2 years ago
- My code repo for random-walks.org☆16Feb 1, 2026Updated 9 months ago
- A Rust crate offering similar functionality to the Python transformers package using Candle.☆15Nov 19, 2024Updated last year
- Inference code for LLaMA models in JAX☆120May 21, 2024Updated 2 years ago
- 🦠 COVID-19 Daily Data from Worldometers with Python☆13Feb 28, 2021Updated 5 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 3 years ago
- Monitoring a PyTorch Lightning CNN with Weights & Biases☆15Jul 26, 2021Updated 5 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.
- Orbax provides common checkpointing and persistence utilities for JAX users☆541Updated this week
- Composable kernels for scikit-learn implemented in JAX.☆47Oct 26, 2020Updated 6 years ago
- PIX is an image processing library in JAX, for JAX.☆452Updated this week
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆28Feb 1, 2025Updated last year
- ☆16Aug 1, 2025Updated last year
- [WIP] A 🔥 interface for running code in the cloud☆87Sep 23, 2026Updated last month
- [ICLR 2026] "Test-Time Scaling in Diffusion LLMs via Hidden Semi-Autoregressive Experts"☆15Feb 10, 2026Updated 8 months ago
- JAX implementation of GPTQ quantization algorithm☆10Jul 19, 2023Updated 3 years ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆25Updated this week
- 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.
- LightningDiT Fork with TREAD implementation☆13Oct 13, 2025Updated last year
- ☆66Aug 2, 2022Updated 4 years ago
- ☆958Oct 7, 2026Updated last month
- ☆26Sep 15, 2022Updated 4 years ago
- Distributed lock backed by Dynamodb☆11Dec 7, 2023Updated 2 years ago
- ☆14Sep 11, 2026Updated last month
- ☆21Updated this week
- Official CLI for deploying and managing sites on Orbiter☆11Dec 31, 2025Updated 10 months ago
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆16Aug 1, 2022Updated 4 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Maximum Entropy-Regularized Multi-Goal Reinforcement Learning (ICML 2019)☆24May 30, 2019Updated 7 years ago
- 🤗 Collection of examples on how to train, deploy and monitor HuggingFace models in Google Cloud Vertex AI☆23Feb 26, 2024Updated 2 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Jan 20, 2024Updated 2 years ago
- Baselines for gymnax 🤖☆82Apr 3, 2023Updated 3 years ago
- ☆12Jun 24, 2025Updated last year
- JAX implementation ViT-VQGAN☆82Sep 21, 2022Updated 4 years ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆41Jul 20, 2026Updated 3 months ago