Flax

Free 0

Neural networks for JAX.

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

0 Reviews

Want to share your thoughts?

Help others discover this tool by sharing it with your professional network.