Similar repositories to jax-ml/oryx:

• jax-ml/oryx github similar
• danistefanovic/build-your-own-x github similar
• lab-ml/annotated_deep_learning_paper_implementations github similar
• pandas-profiling/pandas-profiling github similar
• jeremiecoullon/SGMCMCJax github similar
• cgarciae/treex github similar
• NeilGirdhar/efax github similar
• jax-ml/jax-triton github similar
• cgarciae/treeo github similar
• google/sympy2jax github similar
• hanxiao/bert-as-service github similar
• tuvtran/project-based-learning github similar
• cgarciae/jax_metrics github similar
• mpi4jax/mpi4jax github similar
• google/tree-math github similar
• deepmind/jaxline github similar
• google/jaxtyping github similar
• blackjax-devs/blackjax github similar
• Joshuaalbert/jaxns github similar
• dfm/jpu github similar
• google/trajax github similar
• probml/ssm-jax github similar
• deepmind/distrax github similar
• brentyi/jax_dataclasses github similar
• DifferentiableUniverseInitiative/jax_cosmo github similar
• lab-ml/labml github similar
• danijar/ninjax github similar
• brentyi/jaxlie github similar
• deepmind/dm_pix github similar
• ott-jax/ott github similar
• thomaspinder/GPJax github similar
• google/CommonLoopUtils github similar
• deepmind/kfac-jax github similar
• SymJAX/SymJAX github similar
• deepmind/chex github similar
• srush/learns-dex github similar
• deepmind/dm_aux github similar
• mitmath/18337 github similar
• coax-dev/coax github similar
• henry-prior/jax-rl github similar