Wheelodex » Projects » axlearn

axlearn

View on PyPI — Reverse Dependencies (0)

0.0.1.post20240213000026 axlearn-0.0.1.post20240213000026-py3-none-any.whl
0.0.1 axlearn-0.0.1-py3-none-any.whl

Wheel Details

Project: axlearn
Version: 0.0.1
Filename: axlearn-0.0.1-py3-none-any.whl
Download: [link]
Size: 4652955
MD5: 26ab96783043bb58b1cd8f6b427721cc
SHA256: a4dcdcbababf7c337913136d835b272d4c7efd65ff270c65d29a7abbfa1b86dd
Uploaded: 2023-08-23 00:17:04 +0000

dist-info

METADATA · WHEEL · RECORD · entry_points.txt

METADATA

Metadata-Version: 2.1
Name: axlearn
Version: 0.0.1
Summary: AXLearn
Requires-Python: >=3.8
Requires-Dist: attrs (>=21.3.0)
Requires-Dist: absl-py
Requires-Dist: chex (>=0.1.6)
Requires-Dist: flax (>=0.7.1)
Requires-Dist: importlab (==0.7)
Requires-Dist: jax (==0.4.13)
Requires-Dist: jaxlib (==0.4.13)
Requires-Dist: nltk (==3.7)
Requires-Dist: numpy (<1.24)
Requires-Dist: optax (==0.1.3)
Requires-Dist: portpicker
Requires-Dist: protobuf (==3.20.3)
Requires-Dist: tensorboard-plugin-profile (==2.8.0)
Requires-Dist: tensorflow (==2.8.0); platform_machine == "x86_64"
Requires-Dist: tensorflow-datasets (==4.7.0)
Requires-Dist: tensorflow-io[tensorflow]; platform_machine == "x86_64"
Requires-Dist: tensorflow-io-gcs-filesystem[tensorflow]; platform_machine == "x86_64"
Requires-Dist: tensorflow_text (==2.8.2)
Requires-Dist: tensorstore (>=0.1.21)
Requires-Dist: toml
Requires-Dist: typing-extensions (>=4.2.0)
Requires-Dist: seqio (==0.0.15)
Requires-Dist: attrs (>=21.3.0); extra == "apple-silicon"
Requires-Dist: absl-py; extra == "apple-silicon"
Requires-Dist: chex (>=0.1.6); extra == "apple-silicon"
Requires-Dist: flax (>=0.7.1); extra == "apple-silicon"
Requires-Dist: jax (==0.4.13); extra == "apple-silicon"
Requires-Dist: jaxlib (==0.4.13); extra == "apple-silicon"
Requires-Dist: nltk (==3.7); extra == "apple-silicon"
Requires-Dist: optax (>=0.1.1); extra == "apple-silicon"
Requires-Dist: portpicker; extra == "apple-silicon"
Requires-Dist: tensorboard-plugin-profile (==2.8.0); extra == "apple-silicon"
Requires-Dist: tensorflow-datasets (==4.7.0); extra == "apple-silicon"
Requires-Dist: tensorstore (>=0.1.21); extra == "apple-silicon"
Requires-Dist: typing-extensions (>=4.2.0); extra == "apple-silicon"
Requires-Dist: seqio (==0.0.15); extra == "apple-silicon"
Requires-Dist: black (==23.1a1); extra == "dev"
Requires-Dist: diffusers (==0.16.1); extra == "dev"
Requires-Dist: einops; extra == "dev"
Requires-Dist: evaluate; extra == "dev"
Requires-Dist: fairseq (==0.12.2); extra == "dev"
Requires-Dist: isort; extra == "dev"
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: pycocotools; extra == "dev"
Requires-Dist: pylint (==2.13.9); extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: pytest-xdist; extra == "dev"
Requires-Dist: pytype (==2022.4.22); extra == "dev"
Requires-Dist: scikit-learn (>=1.1.1); extra == "dev"
Requires-Dist: scipy; extra == "dev"
Requires-Dist: sentencepiece (!=0.1.92); extra == "dev"
Requires-Dist: tqdm; extra == "dev"
Requires-Dist: timm (==0.6.12); extra == "dev"
Requires-Dist: torch (>=1.12.1); extra == "dev"
Requires-Dist: torchvision (==0.14.1); extra == "dev"
Requires-Dist: transformers (==4.27.1); extra == "dev"
Requires-Dist: wandb; extra == "dev"
Requires-Dist: wrapt; extra == "dev"
Requires-Dist: cloud-tpu-client; extra == "gcp"
Requires-Dist: crcmod; extra == "gcp"
Requires-Dist: google-api-python-client (==1.8.0); extra == "gcp"
Requires-Dist: google-auth (==1.35.0); extra == "gcp"
Requires-Dist: google-cloud-storage (==2.2.1); extra == "gcp"
Requires-Dist: google-cloud-core (==1.7.3); extra == "gcp"
Requires-Dist: jax[tpu] (==0.4.13); extra == "tpu"
Requires-Dist: setuptools (==65.7.0); extra == "vertexai_tensorboard"
Requires-Dist: google-cloud-aiplatform[tensorboard]; extra == "vertexai_tensorboard"
Provides-Extra: apple-silicon
Provides-Extra: dev
Provides-Extra: gcp
Provides-Extra: tpu
Provides-Extra: vertexai_tensorboard
Description-Content-Type: text/markdown
[Description omitted; length: 1285 characters]

WHEEL

Wheel-Version: 1.0
Generator: flit 3.9.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
axlearn/.DS_Store sha256=_lO8CW1vSLn6F04C--sBEQzHwJ86yL9Qmm7C_V6lWEI 8196
axlearn/__init__.py sha256=5wwrtrl-k1S7ide3oszkxFJ3oqZE2AIomMzB-SgtCRU 47
axlearn/cli/__init__.py sha256=qtmKP_TIZFjxoZXu-ICoIKZJSY6erpmje_36fgCIFLA 2198
axlearn/cli/utils.py sha256=B01As_hbxQcR9ktEj_qQl5_yrLMB-RFPKSSav1PkBcY 11450
axlearn/cli/utils_test.py sha256=A-gRO7_LfyDx5FrypQ5iPZo5UMfDbfWmaMlqDYoLGZQ 12761
axlearn/cli/testdata/dummy.py sha256=xgUsxAT-1rLB3wMfbXVFMxVKWg1e24rATBvWk3P7s3Q 535
axlearn/cloud/__init__.py sha256=UghKI-k_1M9B5BzM-7qm8Eiz5wyY4mLD_zDszEyf4d0 225
axlearn/cloud/common/__init__.py sha256=ZVlK7TPX01OBXHZU3lywiBMXvKbxWqSZk4ijODcdPyo 63
axlearn/cloud/common/bundler.py sha256=MJ9TzLrll-DlBTZ6h4_as1EAYdymDLIWKqU39CgLeTQ 18978
axlearn/cloud/common/bundler_test.py sha256=ofA2u8L7GIMyyOX-ch1P_7qT_NyTGMGl9l7xSn78gx0 6732
axlearn/cloud/common/cleaner.py sha256=R9kJp56qXCMSYNMLwZOg4ImTv7LY2y2fB6iK6zOT84E 498
axlearn/cloud/common/config.py sha256=rPkp0oZK2l32BjAw22aYe51MCS1tsA0ynnoncFVo1qo 12314
axlearn/cloud/common/config_test.py sha256=WQAJTup-G0yCUPTp_xiAO53HkkoZGsecZcDR5iyxaS8 16750
axlearn/cloud/common/docker.py sha256=mzZfOyzbFPZ1svJaIjRT9ocV-ply7XCLjtqtno3IAbc 2098
axlearn/cloud/common/job.py sha256=yxzV-vL3KEygwwf7qlgk-70C9Yko_9WESSuIyQc4WMY 3505
axlearn/cloud/common/job_test.py sha256=9pOwAPTRTKUGA001hTYHOj7k8cgoR1hvMa7nDtB-WLQ 2688
axlearn/cloud/common/quota.py sha256=2CF_KGGwtDmnc2deADFifJ_KxwMDw6GUKiLo6CO9ijg 3014
axlearn/cloud/common/quota_test.py sha256=N24L8JRTm6e5b3S6G3Ibe4scomXJE80V8nA2nJrmqbo 2142
axlearn/cloud/common/scheduler.py sha256=N-FSemgQ2JTvBo1Q2wLl3GPCy2nE_rGqC4zORlOXN-8 16539
axlearn/cloud/common/scheduler_test.py sha256=ack34jBjlQMGCW8dfM3x0zpaxVPPVSnqhyAbBBdhakQ 12530
axlearn/cloud/common/types.py sha256=Nwcoa2pLhYZ7uFCWSwqD-JPcchrweVyRS45sozVw1M8 901
axlearn/cloud/common/utils.py sha256=6Sc7xAejAQvCjxT7oDHFGYn6NVqevaoV6ODDGijM4E0 7102
axlearn/cloud/common/utils_test.py sha256=kBRiKWeTj-OyFH7sfOv9fzcuTaFFlFWqsCPeIvaRk_s 2596
axlearn/cloud/gcp/__init__.py sha256=uCT0BDVQS690nf6_qm2PGXh4tiG4RpEp-x2npKZvQcs 58
axlearn/cloud/gcp/bundler.py sha256=gkmaxunmaFgSMBzfwlLU0XRhNR5ltJWJ8gbly5DJSIA 5948
axlearn/cloud/gcp/bundler_test.py sha256=kPDhuP_viQ2J_LkFrNlnV-e5YB0ylfESYk_ISJMfTpo 3254
axlearn/cloud/gcp/config.py sha256=Y3WFK0ntxujGJIH_V7ZMHW1rOSTXGzIWGf3dfnTdCzg 3345
axlearn/cloud/gcp/config_test.py sha256=qQuyx6cxNHaUC2PuZcc1jCaWqW1e1dJ7yIr-KyyMubM 1734
axlearn/cloud/gcp/job.py sha256=QPGeL4F1nR1fdnlyT_HCvP7cpFz5KxnDkmOWp7E2OYU 10959
axlearn/cloud/gcp/job_test.py sha256=1RKZRO7axdKOsi3QQ37dqB4h_Js9Ani9ge5DRRh13c4 4892
axlearn/cloud/gcp/storage.py sha256=F01fGLrE1pd8dogfKAkdMPSUkYQrEG5WjCPc-W0eFYA 5883
axlearn/cloud/gcp/storage_test.py sha256=0gdJU24OCSlwZ2RATYETsG__H54e0WUYIVUKZZ2mYKk 2162
axlearn/cloud/gcp/test_utils.py sha256=sEl1rAETsY60ZaAi25sX-z-5lP2G4BxKQSvTHMl0shA 586
axlearn/cloud/gcp/tpu.py sha256=2jb2GHB_eDfz6aYRzyPK0minITG0FGb_YPr431lNHpQ 26071
axlearn/cloud/gcp/tpu_cleaner.py sha256=xJ98AYTv6MBRODNfqaI9F3qXs-f-6RPhNo5KuVEbCXE 3930
axlearn/cloud/gcp/tpu_cleaner_test.py sha256=juhdbrBmnSNnXaShG8J8jxY0_I2QF0a9pEWAAyOEBcQ 2539
axlearn/cloud/gcp/tpu_test.py sha256=yuQJc3cMDRCIeYDFPKM656yBrqpPW-THHTC22frkT4E 4817
axlearn/cloud/gcp/utils.py sha256=mOq8Nv8Jkr08NZ1EmCquso7Js6oMhHV10A23g9ToFR8 1905
axlearn/cloud/gcp/utils_test.py sha256=TdjMU9my7kzHu_vphuS9cfNwII8y2L5k1nLZKh8_veE 599
axlearn/cloud/gcp/vertexai_tensorboard.py sha256=ACsT_hm-ABnRXgOfSUKq614hrMDjQ1Ft0nT_p3YqCgY 5557
axlearn/cloud/gcp/vertexai_tensorboard_test.py sha256=PYJmpxXo_4MDFJRIbNNiF4ePUXuA3BJom6ariLJatKw 2149
axlearn/cloud/gcp/vm.py sha256=OrhuH_k4FH2-yLewqP6hcz2Tu-Ox8pgyp7dVi9G0LO4 12531
axlearn/cloud/gcp/vm_test.py sha256=ivzgovklJ2j82ZRZ_Cbj4zzTyddIz9IP-iv0RVzp_-E 1255
axlearn/cloud/gcp/jobs/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
axlearn/cloud/gcp/jobs/bastion_vm.py sha256=RWMrPrf7LoG_IjYrwDFj0FaStcqr9th7RjPgjD1_dDk 49462
axlearn/cloud/gcp/jobs/bastion_vm_test.py sha256=ZMUcpibkqzcEG1LS76ddh-9Gbsl8b2DukydEYjKQn7Y 29946
axlearn/cloud/gcp/jobs/tpu_trainer.py sha256=wA88Ac8GtDmNyBihy2Z1wsDdin9yknTW2yFFnc6-RX0 19794
axlearn/cloud/gcp/jobs/tpu_trainer_test.py sha256=N62EGoJoLt6grPwDNHaDcSrD51AcVCL6mri57C0eRqQ 4767
axlearn/cloud/gcp/jobs/testdata/counter.py sha256=UVmXwhpZp5xuuBA39jrdL0WVdtRVN7qsTuH6CaqrNd0 869
axlearn/cloud/gcp/scripts/start_tpu.sh sha256=9Ji154ot_k38Rpg3z5eSWhhvkIhDs2kmbZA98soUi08 3825
axlearn/cloud/gcp/scripts/start_vm.sh sha256=3UJf42Lxl6UJP5APFVZfozwFLwf_Wo3Bam5sJNAgQ98 3428
axlearn/common/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
axlearn/common/adapter_flax.py sha256=6WJEgcCnacw6sL4FrKM54O9B-JSaE9dqakryB3epcVA 4136
axlearn/common/adapter_flax_test.py sha256=IyPAzS4s7H1piBGq4tU_OW_qPFAH6bD1kaszgNnXvqU 7324
axlearn/common/adapter_torch.py sha256=KHxVd9xp-rZcY036LJkPScVlMFg4vg1czWp0V1viwL4 87599
axlearn/common/adapter_torch_test.py sha256=nbaPLQctVSSrJJWuttXjXKkPKTHqnh1V08BgEXlSlyk 69628
axlearn/common/attention.py sha256=TCP7e46kbIkEvhNL5EVzCJCpxqB383PjZ87ua-CSIr4 130784
axlearn/common/attention_test.py sha256=FQJw8TTUWpr9bfQ4CY4Xa6IrkQ-Veeo3paD5x888yyw 101170
axlearn/common/base_layer.py sha256=9dS5jJcqOa99sjcOeEPeklVCeRefppr6zT-lU54-8RE 16674
axlearn/common/base_layer_test.py sha256=LLq_49RAnANJLQBJOCSlTdS-CXnilv4RGQhEk5X3bu8 18378
axlearn/common/base_model.py sha256=53Unv02QIVTipdPDK0CPUhaAxtz4tFdYSLKPuKXxoGo 1196
axlearn/common/bert.py sha256=fpQ_m6AjgD46qGxnnPcGKKy84sYhlRETQlzu5Z80zmg 20077
axlearn/common/bert_test.py sha256=U824AXZ0ONvLhfuW0PDo3kwIxw7y8OnTOTJ3Q2pf3o0 26888
axlearn/common/causal_lm.py sha256=R9oWee0ZMwyPtaz_ynThxnEH8O-Hgg0DtRWExUo2rqg 15579
axlearn/common/causal_lm_test.py sha256=EMVcx7ZRNaOstCZh770Cu8wrzMqqeepK0WWG2mwS2oo 8918
axlearn/common/checkpointer.py sha256=5nR3ylaPUPyQeHi6Zb3W0kTYob6DlhIpJNoH8BAxLFk 25325
axlearn/common/checkpointer_test.py sha256=ibpgrXPoRRDEHfxuQT1jb27imvxmW8lFJmbn47v9IHE 24475
axlearn/common/config.py sha256=QGwRHGRtP-Ms0gv_KJxBmvkJuPAX721LD45wAYkpHpA 24128
axlearn/common/config_test.py sha256=C_gJ6p_agfmDP_QY7YT41PmnpQnDdchhEQAj4tkr1W8 19599
axlearn/common/conformer.py sha256=iIppDAPxr6ZOtOq98S5YGg0WOPpOjo-QH7t-0cyu_LA 12869
axlearn/common/conformer_test.py sha256=4pNZfPwQXTHQOAaw2zqVHrZRvETwz3tz1E645LNJKJE 11406
axlearn/common/deberta.py sha256=43hn4CT-uY7qXKJxG_PkJt8HivnOSr8D9l6OBObG7Hw 24765
axlearn/common/deberta_test.py sha256=OrwjDu3AHXAbz7V1NfBa08ahIQ9VPO-zNtSx4vWAAw0 25771
axlearn/common/decoder.py sha256=0ilFNqh1ybnWER8eslxKow21LnLdYszCg8NU_qAQ-os 30875
axlearn/common/decoder_test.py sha256=-xqeOjx_43MDsRM-ouKSvzjgHx5WwCwbq11l6yDwXVM 21330
axlearn/common/decoding.py sha256=40rZBvJaklLuPkUydWOe-7z634TdVRmdcjbMs4JqdqA 43746
axlearn/common/decoding_test.py sha256=QvSxXNQATeaiAwf5T0tIc1AGa2ZVRl6xhFYSTlRTJ28 59341
axlearn/common/distilbert.py sha256=6EmfZVqpFsfB1dxOpmMd2BxMWpVTtWiNu8D2w35Fv5M 4150
axlearn/common/distilbert_test.py sha256=CZxLqikx37qfFkieeAchXDR7a5envYyyAsde01qLtPI 3455
axlearn/common/distillation.py sha256=cm8SNneZq83NLqJqkDvpqh5byBnpGDsKlrU_6U0oFRQ 13129
axlearn/common/distillation_test.py sha256=jkDG3MeCNjeExf2TAU7vdjNvsHVMW5Eq1gmee6mIjcA 11226
axlearn/common/dit.py sha256=wXlqKOv23Kipzcos12LcMhOjYooO-gBJ3gBup7bzRtk 16382
axlearn/common/dit_test.py sha256=frnxBVN-ZHfSVnJImvZ22mxOMy2mwwd060MzP_oVQ5E 17974
axlearn/common/embedding.py sha256=Ql2_Dlj9Y5TKY3iPIomXadDGn4gOgeaV-h5NEcJqICc 3996
axlearn/common/embedding_test.py sha256=TWIk7RDKIrCkDnbdEFw1JyV1egOgmCT9x6J6cpcimgQ 4325
axlearn/common/encoder.py sha256=z0FTYQtBi5loIozfABu0g78BNLsUwKxIZyu6AEKD4K4 19809
axlearn/common/encoder_decoder.py sha256=EEdo0_Sx2BoxsukKwhTpZqKWNad3rHXRv7gw4lanGlU 14365
axlearn/common/encoder_decoder_test.py sha256=YGFODqzotgYD-cAJAawN3KBz8ptS1TIAAY3ytarC_qQ 17280
axlearn/common/encoder_test.py sha256=rtJjOMUKqMLw8mpBZ_EWKP3WfLtLGOfySd_6KFJKC04 15722
axlearn/common/eval_classification.py sha256=wEPGQWVIjLnYEqfUxwWp8bwkQpTSGbkuSQqjpTnMcic 6452
axlearn/common/eval_classification_test.py sha256=lxx30RC7hEzbOEvOqTEP2oCo2UdDE-oIPNOVUhLWgLw 9555
axlearn/common/eval_retrieval.py sha256=zoVSZvAwKslN-K6bedjEU5pqohdf1c238QF2kwfuKSE 23741
axlearn/common/eval_retrieval_test.py sha256=SQbon3nrHpwo4ZDGmEMQ_TrGX5Ti2FhvF2osXG8Neg0 18471
axlearn/common/evaler.py sha256=p0m-uRPRRpmd0wiHzXo5FZT16HwL17K5wHeZavUjGaA 29787
axlearn/common/evaler_test.py sha256=5qvdVRc0hrVg9YteJ5y9KPa0t6aGdgGiUpqkikBdznc 27158
axlearn/common/factorized_rms.py sha256=xDwSqA_Sd3JXxsPfLG2LqQNHcCNuXPwNVNIMMack3Ao 9067
axlearn/common/factorized_rms_test.py sha256=IWMqk0ZsaIXbyFuirYdg1V_SGD1XH2NG7AVj81ocyxk 6992
axlearn/common/gda_test.py sha256=nRMjYXV0rf4u803feTO63RQcH4kI08HNTcwkgQUa2o8 2573
axlearn/common/host_array_test.py sha256=G5V0ykZgQXyHMpdrAJLXDbAmOJrXogbcl6aWiEDuFwY 3392
axlearn/common/inference.py sha256=KeQGesEHOMdwthpSBAqDdl-tjkwVhsKDkc46RFhIo-w 15971
axlearn/common/inference_output.py sha256=lwGPO6iPICPSaSqIhcCYwE231mGlnyGLZoi__-pWsvc 8270
axlearn/common/inference_pipeline.py sha256=ReZecENoteLtuR8oVwXtaDt87AuO5cSp0DHf0aYANAQ 5278
axlearn/common/inference_test.py sha256=WmeKutxBOtWkS5EnSrnHZLyTAjCq0-tOD06D0mcIq_E 25879
axlearn/common/input_composite.py sha256=vDEp-d3k0mp61H3pXjmwV_P24nPGNlrqXChY_zmVTdw 2741
axlearn/common/input_composite_test.py sha256=p9kDB6EIemwFw4gCu-_Sp3FbIqs4Kdc5k6COniMFgsg 5978
axlearn/common/input_fake.py sha256=MFddbtFDjUBQKPWk5cfTI-BXeNRs95d1-D0-E9A61D8 13765
axlearn/common/input_glue.py sha256=rxLV3lu1kMf8FCJQOdkPyBpruRJRkemnTVAO3540UKw 29383
axlearn/common/input_glue_test.py sha256=p_w3dgib6G44V9ws9nm--x55aKCUs3rIagryPVCr40o 29676
axlearn/common/input_lm.py sha256=w3GR8TCqTfhe3uuqoSSorUzGkxyxtALkXEVUz0Hcta4 48147
axlearn/common/input_lm_test.py sha256=V4MeKwZ1pTPjy-1MTZjWmUUGIAH5TUqyxh5ZAqML3hE 54781
axlearn/common/input_mlm.py sha256=CKGnm4L62HWwuFgRQ3h4-aoRqMfvvXdp4FuE4tV0-bc 28628
axlearn/common/input_mlm_test.py sha256=LWYWDFPgaHjJQPfN3_RhZPqCiga6KHHzUA8pgdPhKfc 36331
axlearn/common/input_reading_comprehension.py sha256=4PwRHnm22So7wVM7MvgkcF94A-l2cHBHUhzCViZ1z5g 27471
axlearn/common/input_reading_comprehension_test.py sha256=YVSujbzlUlZEHviYCoY4YpC8CEc4cdWYJcbDoPhcCtM 24743
axlearn/common/input_t5.py sha256=t_yZ0odzIK1pGtmfr-JOXCXpI3Xu0D9oo6jbPPsrJ_c 25508
axlearn/common/input_t5_test.py sha256=ffsUxihwYdRxuqnWyuIyTjztotcVWFQe6lW-PQXWfQI 29784
axlearn/common/input_text.py sha256=CUFgkWztMv0VGr128XxLiozXG0uLJ55AROPuU39d96o 22616
axlearn/common/input_text_test.py sha256=qhAx1Cr19oRtIZdchOsyq5Bx3M3LlouI6D_3c7FIlXU 26123
axlearn/common/input_tf_data.py sha256=mzXh-s5roT_HnXG3A4KMq-VwaID4hQ-wL1A-vzQR0ao 42951
axlearn/common/input_tf_data_test.py sha256=h5vgFlqXmQH66jvn1ctwgoWDTo9onWJclECWpr9-mIE 40488
axlearn/common/launch.py sha256=wM1ERWhoW-HVEDKVluRJm4xOAkw7ar6KQ6pAMLCNAgU 5396
axlearn/common/launch_trainer.py sha256=cFyB_Es8_zklWfsIZiAfIkvA5QlQimmxImXCYlR-Ics 2607
axlearn/common/launch_trainer_main.py sha256=gB76l3RADilDhYWiHOWYHZC7m95ns2haTejW3yo3FI8 336
axlearn/common/layers.py sha256=9MqeJTxLfvRvjLynFtUjbFsUHsc6qHG-2MndpJO9qBw 70172
axlearn/common/layers_test.py sha256=YTbV_lToNghPebEZH7sW4u3WAh6pPBQzsIlIHp1xcSc 78379
axlearn/common/learner.py sha256=1wZXlid55dv3tWSxOynxMSuw0qNdo1pGaci9Q_1WGVk 22605
axlearn/common/learner_test.py sha256=waKaZjjj3YNsmByYs1aj8CoBf2hK4mZgBqQnVIq5sbo 31963
axlearn/common/liveness_monitor.py sha256=fQ0245GUfvhQB2x1kS6S0zlRdjRI27-tn-FXRkYttbA 3986
axlearn/common/liveness_monitor_test.py sha256=9fcU7oTxQ_Zol7r-sYNF6xOnqPTJKQOEARQwZcRMbiQ 2876
axlearn/common/logit_modifiers.py sha256=-KhXR45XquSpo8VLTWe_JSXyXflnUvwZ0v8Z2GkMTRc 10871
axlearn/common/logit_modifiers_test.py sha256=xHgtLeDrN-X0XD1zMK5mT6bfC0KdAPD5JPeFmrBaeqs 7358
axlearn/common/lora.py sha256=9fx1mNuHBxXUnvaxV3cX4Eq0DvCCf-WAGp9Z8IBIx4c 18858
axlearn/common/lora_test.py sha256=DzEsJPve33rSMPByI0N4jrOCepSTmrXRxFr6IeYwzNY 16707
axlearn/common/loss.py sha256=VxMujdpSLSchVQLjkbu8cSeuaE-Lb07rkqtBDzdyZ3U 49359
axlearn/common/loss_test.py sha256=SO-wv806fDf-yjbcjF7S_5WX-sKrG7InwTkbGyc500o 43941
axlearn/common/metrics.py sha256=YQ_Zs3Mt_M85dxtH189lT_HwgqC-hwdpRcUZtji3hc4 1902
axlearn/common/metrics_classification.py sha256=vENuS5OzZp--OKl1FpZDiMPKTFDTptKzjvIVNj1EXZI 13753
axlearn/common/metrics_classification_test.py sha256=k2FBYrM54hry09qSJs__sfbI5XicVGlfrhiaPcC-bmc 27558
axlearn/common/metrics_correlation.py sha256=SvI8Z_GYdFTL49DQ-c0mta_gN6yr6ku8R0vcLkw1Hq4 6935
axlearn/common/metrics_correlation_test.py sha256=nPv-ANJhjGQF1NGZyo8OHGBPcgbwQs1hVc6cixK_bbo 5733
axlearn/common/metrics_glue.py sha256=bJdSBVKtgbtdot1snLjK4cbU39VOxw0OM5EuJLV9uCA 6279
axlearn/common/metrics_glue_test.py sha256=CQtavRcz5giUgw2FpiPgTQgtgahLh4BvJmneAgde0ek 9665
axlearn/common/metrics_retrieval.py sha256=A8xIBejyoAd-t_IzpUyfcEJ6r0Mmxpeim_I6Z4S8jkw 20306
axlearn/common/metrics_retrieval_test.py sha256=G3ZLGeMGl-FD2o0lhDcUg08M0asaZzXyMvpsxiuOijw 10467
axlearn/common/metrics_text_dual_encoder.py sha256=ngfrSYoZu6WFVaqx34b8N5t8aFPFixABi9EYa0Hh1kM 6616
axlearn/common/metrics_text_dual_encoder_test.py sha256=V9JGZcSPDZar6Usumi2DPye7pCHR4kzF5FaB1KPANDA 7919
axlearn/common/module.py sha256=2qf1GGn3rbkcppaV_lF2bswk0jutHsm603lmD6Txf-4 27050
axlearn/common/module_test.py sha256=NE-WFOBp8ZaETuSnSTUCk0BWvwqaQNXR9CHq2nvCHFM 22204
axlearn/common/multi_stream_model.py sha256=uhIPr4XkuKrp-Ztbg1Oc-E2TyNzLZjxME5DFqsmvNY0 8507
axlearn/common/multi_stream_model_test.py sha256=7GuWZ7zxLMdZaZBm8TybS0XGPKEQNAQXw-QqL5bc9AM 8808
axlearn/common/multiway_transformer.py sha256=3zzbZcGMC4s7MWTPst9hgQN0wlKZASvCC8D45nvIgqM 20943
axlearn/common/multiway_transformer_test.py sha256=EckpEKB-DG84puxnkZEmfes69CEptFW1WJdFitwz2fw 19199
axlearn/common/neural_retrieval.py sha256=c6Jl3gLSkLvCy6Pzd8xfcBqYQQwI0CyzyFbKwu6-kU4 5691
axlearn/common/neural_retrieval_test.py sha256=ctmGxlLjQ1qBdQ1tKhff1DDEyreV4J_DZndH6f7ESa0 5067
axlearn/common/normalize.py sha256=G0qS7UOYiWdgMAFSmEOCLdBJoP3xoIxAbJoXJNQwQv8 569
axlearn/common/normalize_test.py sha256=tNiPhtZ7QMKxTm18lRxqBZrlwzI3lbpYKxOTiAK3vuk 483
axlearn/common/optimizer_base.py sha256=WDdF6c6x-_gvGoeUHpXPAfqm0cJMOXRV2pTmZkzYa-M 2490
axlearn/common/optimizers.py sha256=Ckx6KH_UQSxoNcFP84nwMaZWNPrMnb0oDWa7HY9ZMns 46206
axlearn/common/optimizers_test.py sha256=g9G0A2NFExn6E-_NzeixY2uCPqAjhFAzvE0W_3oazhQ 37686
axlearn/common/param_converter.py sha256=WhJBUT550D-GKscnsbKlK31AbJe9Yc2mP_VuewJdQd4 90670
axlearn/common/param_converter_test.py sha256=QUAJ3LBCvlHWlmZXrfx2yqYvhx6S-eteKW5SnwT1w7Y 55518
axlearn/common/param_init.py sha256=3xJxh6Oo0l9Y9utN_4QP-O19_M6s5vBVqq8i6RwR4Qw 13052
axlearn/common/param_init_test.py sha256=bW8lx2sIFsgkgt3Dt1o-RpLAVsA7jgSS0KYiyL59rEg 10676
axlearn/common/pipeline.py sha256=4ovJwlR-spjUJGbZvIuotGOfkQAbnkdeYEvTcAbTFQQ 17633
axlearn/common/pipeline_test.py sha256=RrJ5sPwkveSFcgdHzg0DBnHfcdgS8GS7BNr3TRCmwkE 9642
axlearn/common/poolings.py sha256=ZvYzZF3jFa1b435XexSXZmUJaVygQ9GxdP8fEuhcLjU 16765
axlearn/common/poolings_test.py sha256=1FhL9zqIXtlpx0rIOkq3CMmJ5vG58NJWiV15kVbqYG8 19094
axlearn/common/quantizer.py sha256=RGhf2SaYC62rOFG6Ztp8TnVueAAdq68g-3CNtg5DDWk 26060
axlearn/common/quantizer_test.py sha256=DbU6MEZEAl0iB8-P3GZ6AuhIV92zTiYd3kwX9_VHGhQ 35090
axlearn/common/repeat.py sha256=PoNyptVEwuP9WNcLbKYiipb3rA8ryR3RvPCoQMoEgbs 6658
axlearn/common/repeat_test.py sha256=_4Y4WvEGuN4auqQDfTv25pVfk0Uqwy3KK9dSb_PNf9o 13648
axlearn/common/rnn.py sha256=AhOfQXYnfIC2F8kn-soHYEaLtPTL-I5QuhF2o2fuX-s 18068
axlearn/common/rnn_test.py sha256=NtcDYpP7WA4wTqFuEL3pHO6T5rv_k_L9TopEyNpADUU 16290
axlearn/common/schedule.py sha256=aPBk4TCbCTjYXX-TbNPH1YiWRLSZQmu9imb2Dy8tNRg 11059
axlearn/common/schedule_test.py sha256=CAA2oakzTMrE8CpCRi4gPJsV4-vlkIt5DBIMha7RFas 8010
axlearn/common/state_builder.py sha256=YaN0vwgpSzO9HafSZH7IOzfwh4OJ8myH2VYvWdVkHaU 45906
axlearn/common/state_builder_test.py sha256=XGcjBg_uUvT0sz01BJhVa6-ZMxd_73VWdUyCtNc8R_k 45518
axlearn/common/summary_writer.py sha256=yA9S2c7W9b3SDHyc4ga1qrvbx5omYEMnXafn6PLVfe4 11775
axlearn/common/summary_writer_test.py sha256=tS1m598tPTZuax34K7g3LWeB_ttTCqUQyyK24n9BkdU 3764
axlearn/common/t5.py sha256=UjqXyvABDMv1YI4i1DVkBiPBu7K2pJZowuvcxo_tKLw 17322
axlearn/common/t5_test.py sha256=ntvBJS2FdTfQPx0Vp9oCew-K9ClDiuSYC-kiuRNnDGw 16417
axlearn/common/test_utils.py sha256=SPfpKYNNUllRAxi0OXL4FPgwaCy58AhaXh7sJfzLNEU 30683
axlearn/common/text_dual_encoder.py sha256=HQpsVWhH4rsj2XFkzBuGBOXBUARUixFL-Gaj3WKxwVk 19005
axlearn/common/text_dual_encoder_test.py sha256=gs_fPYxUDMMLF-HHz6rnEtiDmFgy6O_4-wkjM9-Ku3w 17290
axlearn/common/text_encoder.py sha256=Q1psCoqqv27ggNc4xJQhXThLZ2Kyu3DGUgqp0I8tK90 2779
axlearn/common/torch_utils.py sha256=-hLGfXnpjIu7_Td-pH0nAxP7bFvuDXM5JQHE9G41Gj4 4477
axlearn/common/trainer.py sha256=NjSBrST_gvo2Y58qMk1WzPT0lZKSSmjrZCD8zJ7kE2A 35237
axlearn/common/trainer_test.py sha256=tBWEj6fYSPXlvCTwvFz-xt1w3pk-8Ex-Kzp7gV9pC4U 31597
axlearn/common/transducer.py sha256=Hm2MNUbhf6Y-mz_rdQixubmKu3njd3nf3-b7MtU4Y2o 26521
axlearn/common/transducer_test.py sha256=y33LYQD1-eEPXgxRix8ghzSTiRqr5DMCyGtZgOSrIkc 26566
axlearn/common/utils.py sha256=L3f183I_PozqxOJXwmQ2T8FizoVfAQ4TS_KfghtKYI8 30372
axlearn/common/utils_spmd.py sha256=2lxqRmPuey-xVqbc5fiNplhqMgZvj5-CBsOjIFR0rNw 4536
axlearn/common/utils_test.py sha256=-gToXBXiEND1IAOms4F3wo60aTpEb_jZrHwdoplqHlw 35835
axlearn/common/utils_text_dual_encoder.py sha256=7At62KLx_ukArKFsKbKZBzwzE8elyf-JreaMly8TXX0 7376
axlearn/common/utils_tf.py sha256=tZWYwmosFXP-06kWIEkj2SsXm4ruAcEvZulL7HK-JIo 790
axlearn/common/utils_tf_test.py sha256=iAZ1Vrx9QP3gnGLH0xGr25Y0Cn4kbMQpEoCNpqo4GIE 959
axlearn/common/vision_transformer.py sha256=XVMo8N98WhkmJghtQkJzkKcSClRsjjY50j0e4sIPFe8 23377
axlearn/common/vision_transformer_test.py sha256=dOLuDY2YE7hgqI5Xc_-YB7yxShMhR1KF-pGc5ibtLLo 12962
axlearn/common/vocabulary_bpe.py sha256=YvY9Ui_KMZt5W3R5EAAgEB-6xMhVHNzPCEf-w2d3tAs 17194
axlearn/common/vocabulary_bpe_test.py sha256=HGepyi0P2xsvSq-StU60CytZMa1ZIjptXqMt_2g5ICc 11576
axlearn/common/flash_attention/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
axlearn/common/flash_attention/attention.py sha256=q3vgd7ynL4AszvjFi9HeihGgInuJkeWF9bEh6mt471Y 22475
axlearn/common/flash_attention/attention_benchmark.py sha256=7dQxZiFTc_0ix8k_m9guxAjgkDXGu75UBu4ZroBDTKs 8894
axlearn/common/flash_attention/attention_test.py sha256=JX4fmUUJMU9pp6AB628QuTZuwBqRZI5CfSD6Sz1LwkA 5027
axlearn/common/flash_attention/layer.py sha256=y6__0DvpRl8fNfOeZnM9KqQahlBGWrqt8xifYlfJvQE 4491
axlearn/common/flash_attention/layer_test.py sha256=Q4G1DMdjrImkthuQASlNCcsGrbttEEnzLasrveaq6ug 8288
axlearn/experiments/.DS_Store sha256=rwnDqitOliTvCr3lO75NyHVkKiRrerNnTmUapFz5BQM 6148
axlearn/experiments/__init__.py sha256=0s0k6hNBOH2XvKDYfQMH8HcRRRmVbvu1-JGXNlWAbHQ 1853
axlearn/experiments/conftest.py sha256=JnvL_gkYAQB3Bhz7r38ZgZOsnFgsid7_QH7mX3SFBJI 380
axlearn/experiments/golden_ckpt_test.py sha256=jqYgDw3IohP2WLHd3VUV3O7QlWBuuREkUzeQ6EXhyFs 3850
axlearn/experiments/golden_config_test.py sha256=qPiE7UKgWq9ZDoEVQTLAUfLtNsgbgaX4F7SnPi4NPTs 10187
axlearn/experiments/trainer_config_utils.py sha256=oXQ1_cMsnQLG1dpJlANCa5peYc-JxF_Z_zUXEFBDF_I 936
axlearn/experiments/trainer_config_utils_test.py sha256=qws5EVzoEt4BlpxSeLvBTARmevcLoQPlYCDWXGvNEVg 1452
axlearn/experiments/testdata/tokenizers/.DS_Store sha256=-2dFzWCzl3CpmdNWQcOJpKfn-C_-DMqrrjPCsgRo4AQ 8196
axlearn/experiments/testdata/tokenizers/bpe/.DS_Store sha256=lFMG5XGOxj0Kip32H_aClZKS2typ3pMGDwjxDH-TypY 6148
axlearn/huggingface/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
axlearn/huggingface/hf_extractive_qa.py sha256=51UCzWnoIKJ0Is4AyBabDyhrAWJhBme4DSH94MAZFqo 8166
axlearn/huggingface/hf_extractive_qa_test.py sha256=NKh_8_ZW2QAuf1QenHVpG0TS00mmuAn2K6kqFa0KkFg 10906
axlearn/huggingface/hf_module.py sha256=l7zDLKOP68h1cMjQ7GvFT2ZxQFcbcCrOKP9uZtPMan8 11360
axlearn/huggingface/hf_sequence_classification.py sha256=D3reUmSYsfR_bL92ACY3Je_svXp1hoW8H8rW84zBrpo 6840
axlearn/huggingface/hf_sequence_classification_test.py sha256=LV1WrUfKIovCTZbRqBsD6egakTitPO_uZu1FLdKXVmA 7870
axlearn/huggingface/hf_text_encoder.py sha256=u5YMad6StdwLOUe4OzJr9Jfh9BugfaqB6GS7hMZFzgI 6103
axlearn/vision/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
axlearn/vision/anchor.py sha256=K97yOgJa1hziFtEpIlPALAVCPh1iSyJR4yxlEfQAifk 12292
axlearn/vision/anchor_test.py sha256=m1i3IXGOuZu6y_PuyocfswdmsTYvVRP0ctUXGY8aDI8 4813
axlearn/vision/attention.py sha256=kwobVLCBZ9axcGjCAb5_xpXb-wWu14jFQ0VGfk_CLiY 15526
axlearn/vision/attention_test.py sha256=uyd78SY4TKrlQWWKdRv0f7xePDlgcPoh1MfMCvNmNHM 6583
axlearn/vision/augment.py sha256=pCHNPmNqXGWEPpt45hyr_QxImfqYNpFvGf9wTeRqeso 39064
axlearn/vision/augment_test.py sha256=sAI8oxe8O1Xy1gORX3Rt_SRQwaAyGLdn6qzPocnlOVo 11358
axlearn/vision/beit_image_tokenizer.py sha256=4zfHMfJRGfhndIQOW1Ck3QVOCMgEFQ8OdmkvP4WhyH4 11305
axlearn/vision/beit_image_tokenizer_test.py sha256=JJ64Q6h3hzqqR5Ih7mqdrGSTBg9qteRpFnmj40Unpy4 3102
axlearn/vision/box_coder.py sha256=PB0Y3s2-7cR2_3CCbNmfVMI-o6rIi37Qx1vbDo3NnR4 6061
axlearn/vision/box_coder_test.py sha256=3ndO4v4QLqUfRq_Y_R-SbCxSIMfB1bW9fzoRryE4ZpI 3875
axlearn/vision/clip.py sha256=1-Hyx13SuN_6SWImFA3i9gn_OENHoVCmMQehmpA6aOQ 24625
axlearn/vision/clip_test.py sha256=Z3FrdY6wuYuw-JlyEe8HE36MgbSof3n23Z4cFzpR0jw 13550
axlearn/vision/coca.py sha256=cSHQkWlPKvg3yy2VRC6aSiyQgQKtqbriJeXO_roi5MU 43696
axlearn/vision/coca_test.py sha256=vl6rMf4oJdMZhL1GX1eqLnqWJKyzI8N7fTllVL2G2sI 23481
axlearn/vision/coco_evaluator.py sha256=fPxWGFoDWCA3oYpStelPgfxc5z8w78NufNRdUjDiaHU 14424
axlearn/vision/coco_utils.py sha256=j5r6w2sng2luv6vGWKP2HxlJPif78-IyxtbA1hbut70 16971
axlearn/vision/coco_utils_test.py sha256=2uErXyuv53VEPPhvyAt96F6pms2Z0AVIvWFBKMhDe2I 6680
axlearn/vision/cyclip.py sha256=3mtO2cWU1Gqn6YJ4cOHH99cco8w4upjM20RjN_Ufjm8 5481
axlearn/vision/cyclip_test.py sha256=o-NM-7zBhga1-l-6F1qTMLrs7HtNdJXX6ipce9lsDfE 3179
axlearn/vision/detection_generator.py sha256=aMyFVtryNOYx6VjTLH4yJ6Wp1laQbKGEIN0BRZ7rrbs 18276
axlearn/vision/detection_generator_test.py sha256=GGgad-muqDDZijda1aY500_9KcABMhWdU3Y2scZiyRk 12694
axlearn/vision/detection_heads.py sha256=pn5ojIwJzz5CX8KVc0Ztd4LQoF5LhWu8DzLgrOzSqM0 12666
axlearn/vision/detection_heads_test.py sha256=ei67ga-dkN3BjTBr4BjqXRbzcgNKY8KChaT4T_wXhRI 5776
axlearn/vision/efficientdet.py sha256=fhMTxqODgcserNF7jmZUcVfmDq2uCWVd6Lnpe-nARt0 15155
axlearn/vision/efficientdet_test.py sha256=n3j37f8G7GJFTYQLpKkJVzEgSkYfuRjWDum04jWGwW0 11031
axlearn/vision/eval_detection.py sha256=T7irGCfeTxhuZ6fRo3wwm89itCOwc5qpBhjI-bNOM8c 5473
axlearn/vision/eval_detection_test.py sha256=AMjmo0IcESOzIbFHlnxyxfI1_H5GHEOvJbF43sUnsUg 5112
axlearn/vision/feature_tokenizer.py sha256=GJVBxkX530Oq0QpA86kIa9MlaAKyI_5aEIpwVwBYCW8 2667
axlearn/vision/feature_tokenizer_test.py sha256=nUG_987EeFqUce6j8L56gxbhoknBx3E8o_Y2cI_ppcE 1528
axlearn/vision/fpn.py sha256=-Q9gkt_-yGzG3t3UkdyUhVvedIGVnvc5Tm7Ey13xDyI 42277
axlearn/vision/fpn_test.py sha256=bPfwQY9-_1MX9lbmPABkpP0vL9QQ26stC_LQfpA6O4Y 14170
axlearn/vision/image_classification.py sha256=BtcoGJNuxjZiIjS7s2qlxbvsm8OmUzUCijyh3EU4hKk 2730
axlearn/vision/image_classification_test.py sha256=XFM2FbUzsduD-ASV05Fat6hWe1t35z1Er0ved8VZjfk 8830
axlearn/vision/input_detection.py sha256=gttYdnbrXv1FP6JJBRwTwFwSOSB5DzITSVM0vTVi5Pw 20306
axlearn/vision/input_detection_test.py sha256=E7qvccEdQN2JX7ocQTefJxsEkx2aMj66X7GbiZkwsBo 4743
axlearn/vision/input_image.py sha256=bOdjcZaEv0OsHIVb-sD5b4QZhpSY52gY_WIiJIr4B4c 16124
axlearn/vision/input_image_test.py sha256=inoYnQFqi-5yvN0_6LA6EZTqjZu0WPlDSXF-oK3eEKE 8387
axlearn/vision/mask_generator.py sha256=heBSxpQ4Gmn_7jZ2qGVne373MUKvpIN5b2_oDS762oI 6843
axlearn/vision/mask_generator_test.py sha256=FCm65x6IXDM1MJi_Vc6Ofqwf7UyN0AgKMBin8pauit0 1206
axlearn/vision/masked_image_model.py sha256=og-Jemr23pPSLT9w22i8Lo4i0C_iO5R1i_zr17_WQIQ 6068
axlearn/vision/masked_image_model_test.py sha256=SkDXR--aMyi7UyIeaG0PLk_03mi2TW-1AKsv7nXWcSU 8126
axlearn/vision/matchers.py sha256=Z5qoVFJgxbO4Vcb08UElEPD8yFxSo7SpqKlH_g-Ic7c 17939
axlearn/vision/matchers_test.py sha256=n1cY1GUWhLj1OhFyXQkZ8pQaAlDht2YWuMTEYF0Gglw 6828
axlearn/vision/metrics_vqa.py sha256=Nqkg-BrrC-iiRgoJWye_2uSoGJXxHrOPhjRl0psM8H8 12714
axlearn/vision/metrics_vqa_test.py sha256=gT-qB6QEoCX072pZdzDKkdpIpXZvDIVEoyGzbLSZQ0E 10815
axlearn/vision/mobilenets.py sha256=QjWK2EUvkg0aGmldL_QA59tYYuV6UDcT3OevrRhgXe4 36441
axlearn/vision/mobilenets_blocks.py sha256=KE9vBAbEXXtMN0Wrqr32sgPZ9t5xdNX338iK7Vz2eqU 14952
axlearn/vision/mobilenets_blocks_test.py sha256=LZfQDORrzKEysz59_GDb1hpBnhSHdd3xfos7kZ9o4D0 6669
axlearn/vision/mobilenets_test.py sha256=MInjY1Qi05n1ttvOpqVBBhcMLgLC3qVrV8i0_0njh0o 12293
axlearn/vision/nms.py sha256=u76Qg3YjEKGAS2NtDAYruwBAAcxnbTvrB5cvm2IRNMM 9014
axlearn/vision/nms_test.py sha256=MXUsjyUTp3LriGHZxrsTb9i00IEpNu8rUG_lSTuqu9s 1261
axlearn/vision/param_converter.py sha256=Z3txyd6lHzXBEkXYkLhtS51I2AptQfJxzcljFJ_5Wj8 21597
axlearn/vision/param_converter_test.py sha256=4plgscb1Tsi3kokU6DbSgGMFhqSnda6VNIKQRJp_DsQ 10493
axlearn/vision/rcnn.py sha256=n-VqLJXHPjJGRvVIWn9CGup962bsg-6ORK7wxvqfJa8 14594
axlearn/vision/rcnn_losses.py sha256=Hdv3SjnhG8eeB4H2khoUvNZHXyaW_MOkO3NOgR_vckU 7381
axlearn/vision/rcnn_losses_test.py sha256=509C3uVkm80J-Y6OdQBmHiirZKMkder4e8VCHFoCR8Q 12527
axlearn/vision/rcnn_sampler.py sha256=KPE9fhZVj-W4OXdmRM6Tyd3QSVyo_93lp-mowfdndgk 7472
axlearn/vision/rcnn_sampler_test.py sha256=VpqHQgV4F4sF2OlZEAPCpZI3KRItwFHeMu9qDJvEaj4 10726
axlearn/vision/rcnn_test.py sha256=_TwsR3s6oDMQl_GAxBp7w0W2dCXLHfKTu7RRbdPMRFc 4588
axlearn/vision/resnet.py sha256=Cb6SZS5Y11x0Bp2k1_aSK6D160PeB_2I2ngp0UJWeiM 21276
axlearn/vision/resnet_test.py sha256=WqHjFaKTX6h633_Q1xA8mbFALUKtj4SKgKiUWwfsN2Y 4049
axlearn/vision/retinanet.py sha256=qKUsVMCsWdFVJ8SUBqxN8K17caN4-TRwE0TMOPSqT8g 22730
axlearn/vision/retinanet_test.py sha256=JyCXdmq3abobFRYw4viwZYEDKbjoE5lxMR0hu9lAHGE 8716
axlearn/vision/roi_aligner.py sha256=NcpLkGvGymgVlAo4EptnKJTlH56f2TCgy0esf2-n47Q 3339
axlearn/vision/roi_aligner_test.py sha256=9b21uEC3wKwItgvI4pyeDI7GCXu2quhvK_zk7C92L7o 2378
axlearn/vision/roi_generator.py sha256=ojR_5tHdj-5n-ZRJkkjdIO9eIew9aIm7G0Ad5J65Z-g 5501
axlearn/vision/roi_generator_test.py sha256=VEMFaU2hcUB0__wDpsjdxE0H8yesw2Sby85m9BPuDxQ 5758
axlearn/vision/rpn_sampler.py sha256=s2u5v-NAGAhuqJPSEd7bqjmWexdSnWeQzg9zTm1AzjA 7574
axlearn/vision/rpn_sampler_test.py sha256=LmEaLh0OjOQxzdVN8X1GcSYQ-mfNAiWqfvmyyDWcswQ 3903
axlearn/vision/samplers.py sha256=44_SviGhSnGN8BfwH1tumHxY-TXn1_5EyszDSYUwZaQ 4352
axlearn/vision/samplers_test.py sha256=kd_2Y9oAOpMa7ZsSgww3sBtqbp4O0ERuGSc_pbmUnG8 9171
axlearn/vision/similarity_ops.py sha256=xOqY6FIuGPRufOgTjtqFIdLd9l50xpsl6iLzT14gHuU 7132
axlearn/vision/similarity_ops_test.py sha256=FOYjpqfFD5gA-Dkl4ncBEb5QGlE2zGY4IQyS1DMCgJ4 5158
axlearn/vision/spatial_transform_ops.py sha256=GecZN8ZNV82HM7J1b3bg8XuW-8cupaYWumldMelNV8o 21753
axlearn/vision/spatial_transform_ops_test.py sha256=3lQ2BSnXq3cd85NWBlHIamh62RwyGUbkZO6CBAAurR0 16798
axlearn/vision/utils_detection.py sha256=vtDplzOLUZSSKH6I8lNAQPxmYs28DnITjo-5YYMylr4 43862
axlearn/vision/utils_detection_test.py sha256=IsQWk4fbqz3-2z3yZG7zuEjoEjvu3jV0mfZwZAQIGNM 15048
axlearn/vision/utils_visualization.py sha256=RLJ-OYYrOtB-OwVhivtu4028ukmqc27ENvaUWlZd32g 4271
axlearn/vision/virtex.py sha256=M6GpTgb_fothJnvJP_jrvelH2F-jrANDGRcBb8_TX-Q 14193
axlearn/vision/virtex_test.py sha256=OmIIYzGUxQAHI8eJMbju11jVhJkpWoiReTdkcD7M1sg 4462
axlearn/vision/vitdet_transformer.py sha256=he89xC0tjT47CKAXxU3vGxLlMxnwlSl9DKliH7ouLj4 9374
axlearn/vision/vitdet_transformer_test.py sha256=r-My9N9tcaD0HFLzLTLosjpnC8eFvdAFMFfhjkGli_Y 6035
axlearn/vision/window_attention.py sha256=7PLVkEwfRQ8SYHYEPUGQ03kwcVjG7foqQNGnIUKr-UM 6362
axlearn/vision/window_attention_test.py sha256=CI_r8-mXNQfbEG2uNUpgs_L9kisTspgS_XDlqpCXkpY 9602
axlearn/vision/imagenet_adversarial_text/README.md sha256=8X0ZEPWNjiB1rmmCLjEpkYC1TnEdjfGFGYv8bMioKTw 1206
axlearn/vision/imagenet_adversarial_text/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
axlearn/vision/imagenet_adversarial_text/add_attack_tfrecord.py sha256=DpQd7Aw_lsezn9IiE-5kU1u7XgtRAYRrYjBRNeaXA0A 4804
axlearn/vision/imagenet_adversarial_text/imagenet-simple.json sha256=N0cA5yLRZgtIUJVuFvop3tPxHaLRBUMc2EeNN1Yk4lQ 25707
axlearn/vision/imagenet_adversarial_text/openai_clip_pred_1tfrecord_target2esti.pickle sha256=bVz39iH9XbX3SMia6_xiHhpdSOJNeawuWAySyvdhFIs 4034875
axlearn/vision/imagenet_adversarial_text/util_im_process.py sha256=YuJTxwdkyL-prDBxmaCw0pYqu5Im_61vk0G4KLeWS7U 1343
axlearn/vision/imagenet_adversarial_text/util_imagenet.py sha256=ZD3bFV2cVHwurYznrZbrvyDSwBFU5GM9auBZpzDAI8w 1021
axlearn/vision/imagenet_adversarial_text/util_tfdata.py sha256=j1Td5TaJ11VccxgnpmVYYghhM8JQLlLsx3IU_YdDZds 964
axlearn/vision/imagenet_adversarial_text/zeroshot_eval_with_opensource_clip.ipynb sha256=gsuLKlJlOVB3gxGSOENIPgwBC8i6ohf56ufyS6eaL_w 218377
axlearn-0.0.1.dist-info/entry_points.txt sha256=JOdFF5nvQb8wzeSXGkAkWPGkX8lTODhGn2r77Yrf-ME 44
axlearn-0.0.1.dist-info/LICENSE sha256=mI7nZCcBfZEGdf1ysA4oSqUdv_o2n0tyEkdEWv78ryc 11311
axlearn-0.0.1.dist-info/WHEEL sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4 81
axlearn-0.0.1.dist-info/METADATA sha256=07VN4HIREXjkmUzw-wIzD-hcOoxjaocn0NmlRIwz1Gs 4671
axlearn-0.0.1.dist-info/RECORD — —

entry_points.txt

[console_scripts]
axlearn = axlearn.cli:main