Similar repositories to ott-jax/ott:

• ott-jax/ott github similar
• danistefanovic/build-your-own-x github similar
• srush/GPU-Puzzles github similar
• hanxiao/bert-as-service github similar
• google-research/ott github similar
• lab-ml/annotated_deep_learning_paper_implementations github similar
• facebookresearch/meta-ot github similar
• thomaspinder/GPJax github similar
• pandas-profiling/pandas-profiling github similar
• cgnorthcutt/cleanlab github similar
• yandex/YaLM-100B github similar
• google/jaxopt github similar
• google/sympy2jax github similar
• patrick-kidger/equinox github similar
• patrick-kidger/diffrax github similar
• RobertTLange/gymnax github similar
• google/tree-math github similar
• pnkraemer/tueplots github similar
• jeremiecoullon/SGMCMCJax github similar
• mitmath/18337 github similar
• probml/pml2-book github similar
• deepmind/distrax github similar
• blackjax-devs/blackjax github similar
• RobertTLange/evosax github similar
• google/CommonLoopUtils github similar
• DifferentiableUniverseInitiative/jax_cosmo github similar
• jacobjinkelly/easy-neural-ode github similar
• google/jaxtyping github similar
• dfm/tinygp github similar
• patrick-kidger/torchcde github similar
• cgarciae/einop github similar
• salesforce/OmniXAI github similar
• GamestonkTerminal/GamestonkTerminal github similar
• jeanfeydy/geomloss github similar
• pytorch/torchdynamo github similar
• cgarciae/treex github similar
• google/evojax github similar
• ChrisWaites/jax-flows github similar
• dfm/extending-jax github similar
• deepmind/dm_pix github similar