flax/flax at main · google/flax · GitHub

8000 flax/flax at main · google/flax · GitHub

Repository navigation

Name		Name	Last commit message	Last commit date
parent directory ..
core		core
experimental		experimental
linen		linen
metrics		metrics
nnx		nnx
oss		oss
testing		testing
training		training
__init__.py		__init__.py
configurations.py		configurations.py
cursor.py		cursor.py
errors.py		errors.py
ids.py		ids.py
io.py		io.py
jax_utils.py		jax_utils.py
py.typed		py.typed
serialization.py		serialization.py
struct.py		struct.py
traceback_util.py		traceback_util.py
traverse_util.py		traverse_util.py
typing.py		typing.py
version.py		version.py

0