-
-
-
-
-
hrm-jax Public
Jax implementation of Hierarchical Reasoning Models (HRM)
-
-
-
jasmine Public
Forked from p-doom/jasmineA simple, performant and scalable JAX-based world modeling codebase
Python Apache License 2.0 UpdatedJul 28, 2025 -
-
-
-
-
-
-
equinox-llama Public
Forked from EleutherAI/equinox-llamaEquinox implementation of llama3 and llama3.1
Python MIT License UpdatedFeb 14, 2025 -
-
nanoRLHF Public
Forked from jackfsuia/nanoRLHFRLHF experiments on a single A100 40G GPU. Support PPO, GRPO, REINFORCE, RAFT, RLOO, ReMax, DeepSeek R1-Zero reproducing.
Python MIT License UpdatedFeb 8, 2025 -
koopman-jax Public
Consistent Koopman Autoencoders
Jupyter Notebook GNU General Public License v3.0 UpdatedFeb 6, 2025 -
-
TinyZero Public
Forked from Jiayi-Pan/TinyZeroClean, minimal, accessible reproduction of DeepSeek R1-Zero
Python Apache License 2.0 UpdatedFeb 1, 2025 -
tiny-grpo Public
Forked from open-thought/tiny-grpoMinimal hackable GRPO implementation
Python Apache License 2.0 UpdatedJan 31, 2025 -
jpc Public
Forked from thebuckleylab/jpcFlexible Inference for Predictive Coding Networks in JAX.
Python MIT License UpdatedDec 6, 2024 -
-
mbrl-lib Public
Forked from facebookresearch/mbrl-libLibrary for Model Based RL
Python MIT License UpdatedOct 10, 2024 -
-
slot-gmm Public
Object centric learning with Gaussian Mixture Models
Jupyter Notebook UpdatedOct 1, 2024 -
kalman_gmm Public
Kalman filters with Gaussian Mixture Model observations
Jupyter Notebook UpdatedOct 1, 2024 -
dreamerv3-torch Public
Forked from NM512/dreamerv3-torchImplementation of Dreamer v3 in pytorch.
Python MIT License UpdatedSep 27, 2024 -
-