jaxlib

Depended on by

• acqdp

• anabel

• auto_uncertainties

• bbo-calibcam

• betanegbinfit

• bottleneck-transformer-flax

• brax

• brax-jumpy

• cfgmdl

• chex

• clu

• cpflow

• d-brax-jumpy

• deltapv

• dibs-lib

• discopy

• dkmri

• dmol-book

• econpizza

• efax

• elba

• evojax

• fastax

• FContin

• fedjax

• flaxmodels

• flowjax

• gdec

• gpmp

• imitation

• itea

• jax-cosmo

• jax_dataclasses

• jax-fid

• jax-finufft

• jax-flows

• jaxlie

• jraph

• kalmantv

• klujax

• local-attention-flax

• madjax

• metax

• mlp-gpt-jax

• mocat

• mpi4jax

• mvtk

• numpyro

• optax

• PaLM-jax

• progen-transformer

• proust

• pychastic

• pyg3

• pzflow

• qgrad

• rA9

• ramsey

• regressor

• renn

• riskparityportfolio

• rlax

• SGMCMCJax

• simphox

• skipatom

• strange

• symjax

• tfc

• timecast

• timecast-nightly

• transformer-in-transformer-flax

• trax

• vit-flax

• xaj