PyRank

Flax Linen Python Packages

Python packages with the GitHub topic flax-linen. Sorted by monthly downloads, with stars and forks.
NITHISHM2410
flax-pilot

A Simplistic trainer for Flax

358 6 0
AshishKumar4
dewml

A nice ML framework based on Jax/Flax. Based on https://github.com/AshishKumar4/FlaxDiff

329 1 0
AshishKumar4
sparxml

Spiky Neural Networks in Jax

238 0 0
AshishKumar4
dew-flash-attn-cu13

A nice ML framework based on Jax/Flax. Based on https://github.com/AshishKumar4/FlaxDiff

166 1 0
AshishKumar4
dew-flash-attn-cu12

A nice ML framework based on Jax/Flax. Based on https://github.com/AshishKumar4/FlaxDiff

151 1 0