Tag · 5 repos
jax
Repositories carrying the jax tag.
Exact tag match5 sample repositories
1240
1294
1915
2298
3341
amathislab/musclemimic
Official repo of Towards Embodied AI with MuscleMimic: Unlocking full-body musculoskeletal motor learning at scale
PythonRobotics & Embodied AIMachine Learning & Data Science
278
40 forks
+1
today
anindex/mtp
Implementation of Model Tensor Planning in JAX, TMLR 2025 & ICLR 2026.
PythonMachine Learning & Data Science
45
2 forks
+1
today
CMSL-HKUST/jax-am
Additive manufacturing simulation with JAX.
Jupyter NotebookMachine Learning & Data Science
361
68 forks
+1
today
EdanToledo/Stoix
🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
PythonMachine Learning & Data ScienceTesting & QA
423
49 forks
+1
today
keras-team/keras
Deep Learning for humans
PythonMachine Learning & Data Science
64.4K
19.8K forks
+1
today