LICENSE
setup.py
aqt/__init__.py
aqt/jax_legacy/__init__.py
aqt/jax_legacy/jax/__init__.py
aqt/jax_legacy/jax/compute_cost_utils.py
aqt/jax_legacy/jax/compute_cost_utils_test.py
aqt/jax_legacy/jax/flax_attention.py
aqt/jax_legacy/jax/flax_attention_test.py
aqt/jax_legacy/jax/flax_layers.py
aqt/jax_legacy/jax/flax_layers_test.py
aqt/jax_legacy/jax/fp_cast.py
aqt/jax_legacy/jax/fp_cast_test.py
aqt/jax_legacy/jax/get_bounds.py
aqt/jax_legacy/jax/get_bounds_test.py
aqt/jax_legacy/jax/hlo_utils.py
aqt/jax_legacy/jax/hlo_utils_test.py
aqt/jax_legacy/jax/primitives.py
aqt/jax_legacy/jax/primitives_test.py
aqt/jax_legacy/jax/quant_config.py
aqt/jax_legacy/jax/quantization.py
aqt/jax_legacy/jax/quantization_test.py
aqt/jax_legacy/jax/shape_utils.py
aqt/jax_legacy/jax/stats.py
aqt/jax_legacy/jax/stats_tag.py
aqt/jax_legacy/jax/stats_tag_test.py
aqt/jax_legacy/jax/stats_test.py
aqt/jax_legacy/jax/test_utils.py
aqt/jax_legacy/jax/train_utils.py
aqt/jax_legacy/jax/train_utils_test.py
aqt/jax_legacy/jax/utils.py
aqt/jax_legacy/jax/flax/__init__.py
aqt/jax_legacy/jax/flax/struct.py
aqt/jax_legacy/jax/imagenet/__init__.py
aqt/jax_legacy/jax/imagenet/check_config_util.py
aqt/jax_legacy/jax/imagenet/configs_test.py
aqt/jax_legacy/jax/imagenet/hparams_config.py
aqt/jax_legacy/jax/imagenet/input_pipeline.py
aqt/jax_legacy/jax/imagenet/models.py
aqt/jax_legacy/jax/imagenet/models_test.py
aqt/jax_legacy/jax/imagenet/pokebnn.py
aqt/jax_legacy/jax/imagenet/pokebnn_test.py
aqt/jax_legacy/jax/imagenet/resnet_cost_test.py
aqt/jax_legacy/jax/imagenet/train.py
aqt/jax_legacy/jax/imagenet/train_test.py
aqt/jax_legacy/jax/imagenet/train_utils.py
aqt/jax_legacy/jax/imagenet/configs/__init__.py
aqt/jax_legacy/jax/imagenet/configs/base_config.py
aqt/jax_legacy/jax/imagenet/configs/resnet101_bfloat16.py
aqt/jax_legacy/jax/imagenet/configs/resnet152_bfloat16.py
aqt/jax_legacy/jax/imagenet/configs/resnet50_w4.py
aqt/jax_legacy/jax/imagenet/configs/resnet50_w4_a2_fixed.py
aqt/jax_legacy/jax/imagenet/configs/resnet50_w4_a4_fixed.py
aqt/jax_legacy/jax/imagenet/configs/resnet50_w4_a4_init8_dense8_fixed.py
aqt/jax_legacy/jax/imagenet/configs/resnet50_w4_a4_init8_dense8_momax.py
aqt/jax_legacy/jax/imagenet/configs/resnet50_w4_a4_init8_momax.py
aqt/jax_legacy/jax/imagenet/configs/resnet50_w4_init8_dense8.py
aqt/jax_legacy/jax/imagenet/configs/resnet50_w8.py
aqt/jax_legacy/jax/imagenet/configs/resnet50_w8_a8_fixed.py
aqt/jax_legacy/jax/imagenet/configs/paper/__init__.py
aqt/jax_legacy/jax/imagenet/configs/paper/resnet50_bfloat16.py
aqt/jax_legacy/jax/imagenet/configs/paper/resnet50_w4_a4_auto.py
aqt/jax_legacy/jax/imagenet/configs/paper/resnet50_w4_a4_init8_dense8_auto.py
aqt/jax_legacy/jax/imagenet/configs/paper/resnet50_w8_a8_auto.py
aqt/jax_legacy/jax/imagenet/configs_script/__init__.py
aqt/jax_legacy/jax/imagenet/configs_script/config_schema.py
aqt/jax_legacy/jax/imagenet/configs_script/config_schema_test.py
aqt/jax_legacy/jax/wmt_mlperf/__init__.py
aqt/jax_legacy/jax/wmt_mlperf/bleu.py
aqt/jax_legacy/jax/wmt_mlperf/decode.py
aqt/jax_legacy/jax/wmt_mlperf/gen_hlo.py
aqt/jax_legacy/jax/wmt_mlperf/input_pipeline.py
aqt/jax_legacy/jax/wmt_mlperf/models.py
aqt/jax_legacy/jax/wmt_mlperf/models_test.py
aqt/jax_legacy/jax/wmt_mlperf/predict.py
aqt/jax_legacy/jax/wmt_mlperf/train.py
aqt/jax_legacy/jax/wmt_mlperf/train_flags.py
aqt/jax_legacy/jax/wmt_mlperf/train_test.py
aqt/jax_legacy/jax/wmt_mlperf/training_hparams.py
aqt/jax_legacy/jax/wmt_mlperf/training_hparams_generator_lib.py
aqt/jax_legacy/jax/wmt_mlperf/training_hparams_test.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_config_scripts/__init__.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_config_scripts/config_schema.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_config_scripts/config_schema_test.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/__init__.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/base_config.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/__init__.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/full_model_4bit_weights_and_auto_acts_reset_stats.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/full_model_8bit_weights_and_auto_acts_reset_stats.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/full_model_8bit_weights_only_fq.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/full_model_8bit_weights_only_int8.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/minimal_model_8bit_weights_and_auto_acts.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/minimal_model_bfloat16.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_2bit_weights_only.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_4bit_weights_and_auto_acts.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_4bit_weights_and_auto_acts_no_logit_sharing.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_4bit_weights_and_fixed_acts.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_4bit_weights_only.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_8bit_weights_and_auto_acts.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_8bit_weights_and_auto_acts_float.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_8bit_weights_and_fixed_acts.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_8bit_weights_only.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_8bit_weights_only_fq.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_bfloat16.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/experimental/small_model_bfloat16_no_logit_sharing.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/leaderboard/__init__.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/leaderboard/full_model_4bit_weights_and_auto_acts.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/leaderboard/full_model_4bit_weights_and_fixed_acts.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/leaderboard/full_model_4bit_weights_only.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/leaderboard/full_model_8bit_weights_and_auto_acts.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/leaderboard/full_model_8bit_weights_and_fixed_acts.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/leaderboard/full_model_8bit_weights_only.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/leaderboard/full_model_bfloat16.py
aqt/jax_legacy/jax/wmt_mlperf/hparams_configs/leaderboard/leaderboard_sweep.py
aqt/jax_legacy/utils/__init__.py
aqt/jax_legacy/utils/analysis_utils.py
aqt/jax_legacy/utils/analysis_utils_test.py
aqt/jax_legacy/utils/common.py
aqt/jax_legacy/utils/config_schema_utils.py
aqt/jax_legacy/utils/config_schema_utils_test.py
aqt/jax_legacy/utils/hparams_utils.py
aqt/jax_legacy/utils/pandas_utils.py
aqt/jax_legacy/utils/pandas_utils_test.py
aqt/jax_legacy/utils/report_utils.py
aqt/jax_legacy/utils/report_utils_test.py
aqt/jax_legacy/utils/summary_utils.py
aqt/jax_legacy/utils/summary_utils_test.py
aqt/jax_legacy/utils/tfevent_utils.py
aqt/jax_legacy/utils/tfevent_utils_test.py
aqtp.egg-info/PKG-INFO
aqtp.egg-info/SOURCES.txt
aqtp.egg-info/dependency_links.txt
aqtp.egg-info/not-zip-safe
aqtp.egg-info/requires.txt
aqtp.egg-info/top_level.txt