README.md
pyproject.toml
flashoptim_jax/__init__.py
flashoptim_jax/adamw.py
flashoptim_jax/compression.py
flashoptim_jax/lion.py
flashoptim_jax/momentum_kernel.py
flashoptim_jax/quantization.py
flashoptim_jax/sgd.py
flashoptim_jax/utils.py
flashoptim_jax.egg-info/PKG-INFO
flashoptim_jax.egg-info/SOURCES.txt
flashoptim_jax.egg-info/dependency_links.txt
flashoptim_jax.egg-info/requires.txt
flashoptim_jax.egg-info/top_level.txt
tests/test_adamw.py
tests/test_checkpointing.py
tests/test_compression.py
tests/test_lion.py
tests/test_optimizer_surface.py
tests/test_quantization.py
tests/test_sgd.py