dogear

enter for all results · esc to close

Praxis

github.comtool

The layer library for Pax with a goal to be usable by other JAX-based ML projects.

visit github.com ↗
0
sign in to dogear this
from
JAX
added
2026-10-10
likes
0

similar

  1. Pax github.com
    tool category #jax link ↗

    A Jax-based machine learning framework for training large scale models.

  2. Praxis github.com
    tool category #microservices link ↗

    Framework for both designing and implementing APIs.

  3. JAX Toolbox github.com
    tool category #jax link ↗

    Nightly CI and optimized examples for JAX on NVIDIA GPUs using libraries such as T5x, Paxml, and Transformer Engine.

  4. RLax github.com
    tool category #jax link ↗

    Library for implementing reinforcement learning agents.

  5. Parallax github.com
    tool category #jax link ↗

    Immutable Torch Modules for JAX.

  6. jax-models github.com
    tool category #jax link ↗

    Implementations of research papers originally without code or code written with frameworks other than JAX.

JAX › Libraries > New Libraries: “The layer library for Pax with a goal to be usable by other JAX-based ML projects.”