Hardware accelerated, batchable and differentiable optimizers in JAX.
created at July 12, 2021, 5:16 p.m.
Mathematical operations for JAX pytrees
created at Dec. 18, 2021, 1:56 a.m.
USB Keystroke Injection Protection
created at Feb. 26, 2020, 5:31 p.m.
A simple, performant and scalable Jax LLM!
created at Feb. 28, 2023, 7:47 p.m.
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
created at June 14, 2022, 4:04 p.m.