.gitattributes
.gitignore
.pre-commit-config.yaml
LICENSE
README.md
pyproject.toml
setup.cfg
test-requirements.txt
.github/workflows/tests.yaml
docs/design.md
docs/operators.md
examples/scalify-quickstart.ipynb
examples/cifar10/cifar10_training.py
examples/cifar10/cifar10_training_with_optax.py
examples/cifar10/dataset_cifar10.py
examples/mnist/datasets.py
examples/mnist/mnist_classifier_from_scratch.py
examples/mnist/mnist_classifier_from_scratch_fp8.py
examples/mnist/mnist_classifier_mlp_flax.py
examples/mnist/flax/README.md
examples/mnist/flax/main.py
examples/mnist/flax/requirements.txt
examples/mnist/flax/train.py
examples/mnist/flax/configs/__init__.py
examples/mnist/flax/configs/default.py
jax_scalify/__init__.py
jax_scalify/_version.py
jax_scalify.egg-info/PKG-INFO
jax_scalify.egg-info/SOURCES.txt
jax_scalify.egg-info/dependency_links.txt
jax_scalify.egg-info/requires.txt
jax_scalify.egg-info/top_level.txt
jax_scalify/core/__init__.py
jax_scalify/core/datatype.py
jax_scalify/core/debug.py
jax_scalify/core/interpreters.py
jax_scalify/core/pow2.py
jax_scalify/core/typing.py
jax_scalify/core/utils.py
jax_scalify/lax/__init__.py
jax_scalify/lax/base_scaling_primitives.py
jax_scalify/lax/scaled_ops_common.py
jax_scalify/lax/scaled_ops_l2.py
jax_scalify/ops/__init__.py
jax_scalify/ops/cast.py
jax_scalify/ops/debug.py
jax_scalify/ops/rescaling.py
jax_scalify/ops/utils.py
jax_scalify/tree/__init__.py
jax_scalify/tree/tree_util.py
tests/core/test_datatype.py
tests/core/test_interpreter.py
tests/core/test_pow2.py
tests/core/test_utils.py
tests/lax/test_base_scaling_primitives.py
tests/lax/test_numpy_integration.py
tests/lax/test_scaled_ops_common.py
tests/lax/test_scaled_ops_l2.py
tests/lax/test_scipy_integration.py
tests/ops/test_cast.py
tests/ops/test_debug.py
tests/ops/test_rescaling.py
tests/tree/test_tree_util.py