deepmind/haiku
JAX-based neural network library
Language: Python
#deep_learning #deep_neural_networks #jax #machine_learning #neural_networks
Stars: 212 Issues: 4 Forks: 8
https://github.com/deepmind/haiku

google/jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
Language: Python
#optimization #jax
Stars: 171 Issues: 0 Forks: 0
https://github.com/google/jaxopt

samuela/git-re-basin
Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"
Language: Python
#deep_learning #deeplearning #jax #machine_learning #neural_networks
Stars: 181 Issues: 0 Forks: 10
https://github.com/samuela/git-re-basin

apple/ml-aim
This repository provides the code and model checkpoints of the research paper: Scalable Pre-training of Large Autoregressive Image Models
Language: Python
#jax #large_scale_vision_models #mlx #pytorch
Stars: 181 Issues: 1 Forks: 8
https://github.com/apple/ml-aim