Skip to content
@jax-ml

jax-ml

miscellaneous libraries and projects relating to JAX

Popular repositories Loading

  1. jax-triton jax-triton Public

    jax-triton contains integrations between JAX and OpenAI Triton

    Python 311 30

  2. oryx oryx Public

    Oryx is a library for probabilistic programming and deep learning built on top of Jax.

    Python 201 9

  3. ml_dtypes ml_dtypes Public

    A stand-alone implementation of several NumPy dtype extensions used in machine learning.

    C++ 134 19

  4. bayeux bayeux Public

    State of the art inference for your bayesian models.

    Python 104 4

  5. coix coix Public

    Inference Combinators in JAX

    Jupyter Notebook 41 2

  6. australis australis Public archive

    C++ 29 4

Repositories

Showing 9 of 9 repositories
  • ml_dtypes Public

    A stand-alone implementation of several NumPy dtype extensions used in machine learning.

    jax-ml/ml_dtypes’s past year of commit activity
    C++ 134 Apache-2.0 19 17 4 Updated Jul 23, 2024
  • jax-tpu-embedding Public
    jax-ml/jax-tpu-embedding’s past year of commit activity
    Python 6 Apache-2.0 1 0 9 Updated Jul 19, 2024
  • jax-triton Public

    jax-triton contains integrations between JAX and OpenAI Triton

    jax-ml/jax-triton’s past year of commit activity
    Python 311 Apache-2.0 30 35 (1 issue needs help) 17 Updated Jul 10, 2024
  • bayeux Public

    State of the art inference for your bayesian models.

    jax-ml/bayeux’s past year of commit activity
    Python 104 Apache-2.0 4 2 0 Updated Jul 10, 2024
  • oryx Public

    Oryx is a library for probabilistic programming and deep learning built on top of Jax.

    jax-ml/oryx’s past year of commit activity
    Python 201 Apache-2.0 9 11 (1 issue needs help) 1 Updated Jun 27, 2024
  • coix Public

    Inference Combinators in JAX

    jax-ml/coix’s past year of commit activity
    Jupyter Notebook 41 Apache-2.0 2 8 0 Updated Jun 24, 2024
  • .allstar Public
    jax-ml/.allstar’s past year of commit activity
    0 0 0 0 Updated Dec 6, 2022
  • australis Public archive
    jax-ml/australis’s past year of commit activity
    C++ 29 Apache-2.0 4 0 0 Updated Nov 18, 2022
  • .github Public archive
    jax-ml/.github’s past year of commit activity
    0 0 0 0 Updated Oct 26, 2022

Most used topics

Loading…