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

kyegomez/OpenMythos
A theoretical reconstruction of the Claude Mythos architecture, built from first principles using the available research literature.
Language: Python
#ai #anthropic #attention #claude #claude_ai #claude_code #claude_code_plugin #claude_mythos #claude_sonnet #deepmind #gpt_5 #gpt_7 #jax #looped_transformers #ml #pytorch #sonnet #torch
Stars: 2024 Issues: 7 Forks: 373
https://github.com/kyegomez/OpenMythos