Skip to content
@jax-ml

jax-ml

Pushing back the limits on numerical computing.

Pinned Loading

  1. jax jax Public

    Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

    Python 30.5k 2.8k

  2. jax-triton jax-triton Public

    jax-triton contains integrations between JAX and OpenAI Triton

    Python 341 37

  3. ml_dtypes ml_dtypes Public

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

    C++ 209 29

Repositories

Showing 10 of 11 repositories
  • jax Public

    Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

    jax-ml/jax’s past year of commit activity
    Python 30,456 Apache-2.0 2,799 1,433 (5 issues need help) 412 Updated Nov 11, 2024
  • jax-ml/jax-ai-stack’s past year of commit activity
    Python 66 Apache-2.0 13 1 4 Updated Nov 11, 2024
  • 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++ 209 Apache-2.0 29 22 6 Updated Nov 7, 2024
  • jax-triton Public

    jax-triton contains integrations between JAX and OpenAI Triton

    jax-ml/jax-triton’s past year of commit activity
    Python 341 Apache-2.0 37 36 (1 issue needs help) 17 Updated Oct 31, 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 219 Apache-2.0 10 12 (1 issue needs help) 2 Updated Oct 29, 2024
  • coix Public

    Inference Combinators in JAX

    jax-ml/coix’s past year of commit activity
    Jupyter Notebook 43 Apache-2.0 2 8 0 Updated Oct 24, 2024
  • jax-ml/jax-tpu-embedding’s past year of commit activity
    Python 8 Apache-2.0 2 0 9 Updated Oct 11, 2024
  • bayeux Public

    State of the art inference for your bayesian models.

    jax-ml/bayeux’s past year of commit activity
    Python 164 Apache-2.0 6 3 1 Updated Oct 1, 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 17, 2022

Most used topics

Loading…