neural-tangents in n2cholas/awesome-jax

Fast and Easy Infinite Neural Networks in Python

updated at Nov. 13, 2024, 12:37 p.m.

62 +0

2,278 +1

226 +0

evojax in n2cholas/awesome-jax

None

updated at Nov. 14, 2024, 10:02 a.m.

23 +0

842 +2

87 +0

flax in krzjoa/awesome-python-data-science, n2cholas/awesome-jax

Flax is a neural network library for JAX that is designed for flexibility.

updated at Nov. 16, 2024, 2:13 p.m.

86 +0

6,136 +27

646 +1

brax in n2cholas/awesome-jax

Massively parallel rigidbody physics simulation on accelerator hardware.

updated at Nov. 17, 2024, 1:30 a.m.

34 +0

2,342 +15

255 +1

dopamine in Ly0n/awesome-robotic-tooling, krzjoa/awesome-python-data-science

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.

updated at Nov. 17, 2024, 4:46 a.m.

425 +0

10,569 +8

1,375 -2