LoRA for arbitrary JAX models and functions
created at April 21, 2023, 11:50 a.m.
Really Fast End-to-End Jax RL Implementations
created at Feb. 25, 2023, 3:38 p.m.
Add a tqdm progress bar to your JAX scans and loops.
created at Jan. 16, 2023, 4:54 p.m.
JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et al.
created at Jan. 4, 2023, 11:09 p.m.
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
created at Dec. 21, 2022, 9:20 a.m.
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
created at Nov. 22, 2022, 12:55 p.m.
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
created at Aug. 11, 2022, 7:34 a.m.
GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
created at Aug. 8, 2022, 5 p.m.
A Python package of computer vision models for the Equinox ecosystem.
created at July 24, 2022, 10:02 p.m.
SPU (Secure Processing Unit) aims to be a provable, measurable secure computation device, which provides computation ability while keeping your private data protected.
created at July 2, 2022, 9:05 a.m.