README.md
pyproject.toml
setup.py
flarejax/__init__.py
flarejax.egg-info/PKG-INFO
flarejax.egg-info/SOURCES.txt
flarejax.egg-info/dependency_links.txt
flarejax.egg-info/requires.txt
flarejax.egg-info/top_level.txt
flarejax/flr/_filter.py
flarejax/flr/_lookup.py
flarejax/flr/_module.py
flarejax/flr/_random.py
flarejax/flr/_serial.py
flarejax/flr/_tcheck.py
flarejax/flr/_transform.py
flarejax/net/__init__.py
flarejax/net/_attention.py
flarejax/net/_combine.py
flarejax/net/_einops.py
flarejax/net/_linear.py
flarejax/net/_norm.py
flarejax/opt/__init__.py
flarejax/opt/_adam.py
flarejax/opt/_opt.py
flarejax/opt/_schedule.py
flarejax/opt/_sgd.py