Similar repositories to google/learned_optimization:

• google/learned_optimization github similar
• danistefanovic/build-your-own-x github similar
• lab-ml/annotated_deep_learning_paper_implementations github similar
• tuvtran/project-based-learning github similar
• hanxiao/bert-as-service github similar
• pandas-profiling/pandas-profiling github similar
• ReactTraining/react-router github similar
• lab-ml/labml github similar
• RobertTLange/evosax github similar
• DarshanDeshpande/jax-models github similar
• lucidrains/PaLM-jax github similar
• cgnorthcutt/cleanlab github similar
• GamestonkTerminal/GamestonkTerminal github similar
• zhiqi-li/BEVFormer github similar
• mpi4jax/mpi4jax github similar
• google/jaxopt github similar
• RobertTLange/gymnax github similar
• ethanluoyc/magi github similar
• google/CommonLoopUtils github similar
• google/tree-math github similar
• sannykim/transformers github similar
• meilisearch/MeiliSearch github similar
• RH12503/triangula github similar
• k6io/k6 github similar
• mle-infrastructure/mle-hyperopt github similar
• rwightman/efficientnet-jax github similar
• jeremiecoullon/SGMCMCJax github similar
• kingoflolz/CLIP_JAX github similar
• lfbo-ml/lfbo github similar
• lucidrains/mlp-gpt-jax github similar
• mitmath/18337 github similar
• dfm/extending-jax github similar
• cgarciae/treeo github similar
• alonfnt/bayex github similar
• deepmind/kfac-jax github similar
• google/sympy2jax github similar
• lucidrains/hourglass-transformer-pytorch github similar
• google/spectral-density github similar
• thomaspinder/GPJax github similar
• microsoft/varuna github similar