jaxcld/__init__.py,sha256=Uhz_kKDd4fzZT-4BcYfg4eo8E8n_PqkAcrHChqp4Aug,1445
jaxcld/models/__init__.py,sha256=MffMZ2jQ8RDWJSdersxQ-5-nN6WHJZwqo9JZAHMBAGM,325
jaxcld/models/asr_model.py,sha256=oOD1O9YDuYZS5oRlMTgTfqpZvov1VrpcUT5H3amMB7I,19808
jaxcld/models/cvx_grelu_mlp.py,sha256=R1PpUaV3E8gkzDnCBNEZVvSJbhG5gvJ6oVj5XthMhdo,2341
jaxcld/models/cvx_mlp.py,sha256=kX4eu7bdqUb1MhXrGOxOhXqPG9shFgUD__-iL6ThExw,1173
jaxcld/models/cvx_relu_mlp.py,sha256=PMuXUIelRQj_HuqLoTianze7RO2EADSU5Yr0IBjltj0,3873
jaxcld/models/get_model.py,sha256=noxDvln1-XyJ6RnDmIXqeFOqtaNeY-1CjSm1pFBiLOY,814
jaxcld/models/grelu_mlp.py,sha256=R9eobX8rCsmAS1aTuuSat5vfzHFxeNRrs5R5Gbv0Sms,1065
jaxcld/models/lang_detect_head.py,sha256=noIOKSji9LEhME_j8gonuINexzd6WJZidc62cYzab7Y,5532
jaxcld/models/relu_mlp.py,sha256=3CEGWvMycADXqs0Y2JcahVDrr_KmUUdjILaQ8FOPlmM,1802
jaxcld/models/two_layer_mlp.py,sha256=-1OURhhs6dZTBxy7scme85TyHgbAtobpgeOd0amonU0,258
jaxcld/optimizers/__init__.py,sha256=esZL-GVHMI4IzYk3fQrdR6r-u-Mhxz3WfNpRHpmHGZY,72
jaxcld/optimizers/adamW.py,sha256=PEm81lhxkbU5FCOS4VlcwnoIwYmJII8te5hzYWFe3o4,1179
jaxcld/optimizers/admm.py,sha256=ReonKa-MKjJyahtOyWb4S3cjcLufSVb_ulgmORXKoLc,3818
jaxcld/optimizers/dadapt_adamW.py,sha256=fICkCOREg7wyLRdFK-TKUiRUin2K5NuwoxAjElujXIg,1210
jaxcld/optimizers/pcg.py,sha256=g8ai2GGJ9espKCgif5Kqfq0YQGH6bR6UBvW_c8E0X6M,1561
jaxcld/optimizers/sgd.py,sha256=vYSDf_TNsk4uxZLlGERSh6bZ_S2iytGf41NAZk8VVZQ,1118
jaxcld/optimizers/shampoo.py,sha256=P6AvcU8tkqCe7qzSxK97fXm7ZYPU0wl8CbUHsFPYTDg,1115
jaxcld/optimizers/yogi.py,sha256=CrZCq8RhzgHBONQ0FL5I5kJ7UpNuU4jB7FVQUpBX6hM,985
jaxcld/optimizers/dist_shampoo/__init__.py,sha256=xD87uSdp8xHPMjxOfSxsMwG-t9Fh7ce15EWaCkRl7Vs,63
jaxcld/optimizers/dist_shampoo/distributed_shampoo.py,sha256=WaQKUWZDGGduX7bR-Vc--uXwH3y8QNDlZgz8MHKwgvs,111865
jaxcld/optimizers/dist_shampoo/quantization_utils.py,sha256=3zwQCCkE5_ZOwpPaQlyG-XpomJau-RScjtIih_GrVwI,4247
jaxcld/preconditioner/__init__.py,sha256=uB9ntrEG5X1jfXoD1pOTlBCBdWw365AWkda8tUIvoJE,62
jaxcld/preconditioner/nystrom.py,sha256=o0fTs_XXijUd7H4l0a8l_dKvL-XH1c44KDn_eZKJ8po,3233
jaxcld/training/__init__.py,sha256=a83d5fwwkDiru4bdsUQ5coFMl7l-OjyiQW6YBMY_m5w,111
jaxcld/training/train.py,sha256=yNLcM8ahxW68EKOpUL6K_nH8bWQsosv36FyTMzLT2xA,5509
jaxcld/training/train_no_jit.py,sha256=QNO39sxxwMf-wWmuh3-fkzUHI11LiKvOaMweh3TM9CE,4006
jaxcld/utils/__init__.py,sha256=1peLYEv74vBLiJCO7NKhNs9rYh7YBCJsEodpI7l_P9I,79
jaxcld/utils/linops_utils.py,sha256=sgTZlxWK8vbR6vnVfMWAOR8W5J7WCpmvPKXO0qgajRw,1132
jaxcld/utils/load_data.py,sha256=hdjBogTJxGtXgdaQqRZQcm4VEOnw5IwokZw3qFl0Bc0,16932
jaxcld/utils/metric_utils.py,sha256=sf2IqeIdXAjnWV_c7ZMRRk4UkAiipVVdMIgWSxQp_qY,2069
jaxcld/utils/model_utils.py,sha256=wwr0oUTaBE63_mVvpoMk3ZUoiCjpVjQ_BfiSmOqS0E4,4149
jaxcld/utils/opt_utils.py,sha256=6C-bMz7K_1GBYTtgpjd70Fd2_Wq2Yy7l_BamrcbygoM,914
jaxcld/utils/proximal_utils.py,sha256=jOLBGlb4Gdq7nyw4FFbXehiid82qtk05rrqNiYYPJz4,578
jaxcld/utils/train_utils.py,sha256=n6N7hn-lhumZ-J4giSv63Wc4-XXUX-gm4hvFOmSFnLg,215
jaxcld/utils/whisper_dataloader.py,sha256=J5QP2Vc4J-av4r2_e8doiDgQtDtVoaroAZ5k-W77HgA,5129
jaxcld-0.1.0.dist-info/METADATA,sha256=9yeqaoY6KiEnWkqW4Tlfn3WIwZoVTO8d_VVAZUD9mo4,2781
jaxcld-0.1.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
jaxcld-0.1.0.dist-info/top_level.txt,sha256=cFcH-qCj2LoqpbR8RUv5k-dFEnD07Z6nX0BE-MuUbto,7
jaxcld-0.1.0.dist-info/RECORD,,
