Jonathan-Shoemaker's Stars
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
fastmachinelearning/hls4ml
Machine learning on FPGAs using HLS