Flax is a neural network library for JAX
A functional programming toolkit for R
A JAX-native LLM Post-Training Library
Composable transformations of Python+NumPy programs
A Graph Neural Network Library in Jax
ERNE - Extended Randomized Numerical alignEr
Discovery, Scanning, and Fingerprinting via Broadcast and Multicast Tr
parallelized pmap
C\C++ utilities