Skip to content

@jax-ml jax-ml

Change the repository type filter

All

    Repositories list

    11 repositories

    • jax

      Public
      Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
      jax
      Python
      •
      Apache License 2.0
      •2.8k•30k•1.4k•401•Updated Oct 23, 2024Oct 23, 2024

    • jax-triton

      Public
      jax-triton contains integrations between JAX and OpenAI Triton
      Python
      •
      Apache License 2.0
      •35•337•36•18•Updated Oct 22, 2024Oct 22, 2024

    • oryx

      Public
      Oryx is a library for probabilistic programming and deep learning built on top of Jax.
      probabilistic-programmingjaxpython
      Python
      •
      Apache License 2.0
      •10•217•12•2•Updated Oct 22, 2024Oct 22, 2024

    • jax-ai-stack

      Public
      Python
      •
      Apache License 2.0
      •10•57•0•3•Updated Oct 21, 2024Oct 21, 2024

    • ml_dtypes

      Public
      A stand-alone implementation of several NumPy dtype extensions used in machine learning.
      C++
      •
      Apache License 2.0
      •28•203•21•7•Updated Oct 14, 2024Oct 14, 2024

    • jax-tpu-embedding

      Public
      Python
      •
      Apache License 2.0
      •2•8•0•9•Updated Oct 11, 2024Oct 11, 2024

    • bayeux

      Public
      State of the art inference for your bayesian models.
      Python
      •
      Apache License 2.0
      •6•154•3•1•Updated Oct 1, 2024Oct 1, 2024

    • coix

      Public
      Inference Combinators in JAX
      Jupyter Notebook
      •
      Apache License 2.0
      •2•43•8•0•Updated Jun 24, 2024Jun 24, 2024

    • .allstar

      Public
      0•0•0•0•Updated Dec 6, 2022Dec 6, 2022

    • australis

      Public archive
      C++
      •
      Apache License 2.0
      •4•29•0•0•Updated Nov 18, 2022Nov 18, 2022

    • .github

      Public archive
      0•0•0•0•Updated Oct 26, 2022Oct 26, 2022