jax

Depended on by

• acqdp

• alphafold

• alphafold2mmopt

• alphafold-colabfold

• alphafold-kagglefold

• anabel

• augmax

• auto_uncertainties

• bax

• bbo-calibcam

• betanegbinfit

• bigbench

• bottleneck-transformer-flax

• brainpy

• brain-py

• brax

• brax-jumpy

• cfgmdl

• chex

• clu

• cpflow

• d-brax-jumpy

• deepvit-flax

• deltapv

• dibs-lib

• diffmah

• discopy

• dkmri

• dmol-book

• dsps

• econpizza

• efax

• elba

• evojax

• fadge

• fastax

• FContin

• fedjax

• flaxmodels

• flowjax

• fnet-flax

• funfact

• gdec

• gmlp-flax

• gpax

• gpmp

• graphmlp-flax

• hypernn

• hyper-nn

• imitation

• itea

• jax2torch

• jax-cosmo

• jax_dataclasses

• jax-fid

• jax-finufft

• jax-flows

• jaxlie

• jaxrk

• jb-everything

• jimm

• jraph

• kalmantv

• klujax

• local-attention-flax

• madjax

• MeerKATgen

• metax

• mlp-gpt-jax

• mlpmixer-flax

• mocat

• mpi4jax

• mvtk

• numpyro

• optax

• PaLM-jax

• perceiver-jax

• progen-transformer

• proust

• pychastic

• pyg3

• pzflow

• rA9

• ramsey

• regressor

• renn

• resmlp-flax

• riskparityportfolio

• rlax

• SGMCMCJax

• shamnet

• simphox

• skipatom

• strange

• symjax

• tfc

• timecast

• timecast-nightly

• TorchOpt

• transformer-in-transformer-flax

• trax

• ttax

• vanilla-transformer-jax

• vit-flax

• xaj