Giters

steinst / flax

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

This repository is not active

About

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

License:Apache License 2.0

Languages

Language:Python 97.8%Language:Jupyter Notebook 1.5%Language:Shell 0.3%Language:Dockerfile 0.3%