README.md
README_PIP.md
pyproject.toml
jaxcld/__init__.py
jaxcld.egg-info/PKG-INFO
jaxcld.egg-info/SOURCES.txt
jaxcld.egg-info/dependency_links.txt
jaxcld.egg-info/requires.txt
jaxcld.egg-info/top_level.txt
jaxcld/models/__init__.py
jaxcld/models/asr_model.py
jaxcld/models/cvx_grelu_mlp.py
jaxcld/models/cvx_mlp.py
jaxcld/models/cvx_relu_mlp.py
jaxcld/models/get_model.py
jaxcld/models/grelu_mlp.py
jaxcld/models/lang_detect_head.py
jaxcld/models/relu_mlp.py
jaxcld/models/two_layer_mlp.py
jaxcld/optimizers/__init__.py
jaxcld/optimizers/adamW.py
jaxcld/optimizers/admm.py
jaxcld/optimizers/dadapt_adamW.py
jaxcld/optimizers/pcg.py
jaxcld/optimizers/sgd.py
jaxcld/optimizers/shampoo.py
jaxcld/optimizers/yogi.py
jaxcld/optimizers/dist_shampoo/__init__.py
jaxcld/optimizers/dist_shampoo/distributed_shampoo.py
jaxcld/optimizers/dist_shampoo/quantization_utils.py
jaxcld/preconditioner/__init__.py
jaxcld/preconditioner/nystrom.py
jaxcld/training/__init__.py
jaxcld/training/train.py
jaxcld/training/train_no_jit.py
jaxcld/utils/__init__.py
jaxcld/utils/linops_utils.py
jaxcld/utils/load_data.py
jaxcld/utils/metric_utils.py
jaxcld/utils/model_utils.py
jaxcld/utils/opt_utils.py
jaxcld/utils/proximal_utils.py
jaxcld/utils/train_utils.py
jaxcld/utils/whisper_dataloader.py
tests/test_final_dry_asr_and_heads.py