EasyDeL

View on PyPIReverse Dependencies (0)

0.0.80 easydel-0.0.80-py3-none-any.whl

Wheel Details

Project: EasyDeL
Version: 0.0.80
Filename: easydel-0.0.80-py3-none-any.whl
Download: [link]
Size: 711773
MD5: b8fef9a2e4512f1ac25551bad23d37a4
SHA256: b699f55dc766159fb322bd4b7c9165e119bb67f6d637e18405993b003db9202d
Uploaded: 2024-12-04 15:22:08 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: easydel
Version: 0.0.80
Summary: An open-source library to make training faster and more optimized in JAX
Author: Erfan Zare Chavoshi
Author-Email: Erfanzare810[at]gmail.com
Home-Page: https://github.com/erfanzar/EasyDeL
Project-Url: Documentation, https://erfanzar.github.io/EasyDeL
Project-Url: Repository, https://github.com/erfanzar/EasyDeL
License: Apache-2.0
Keywords: JAX,CUDA,Deep Learning,Machine Learning,XLA
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10,<3.13
Requires-Dist: datasets (<3.0.0,>=2.19.2)
Requires-Dist: fastapi (<0.116.0,>=0.115.2)
Requires-Dist: fjformer (==0.0.91)
Requires-Dist: ipython (<9.0.0,>=8.17.2)
Requires-Dist: jax (>=0.4.34)
Requires-Dist: jaxlib (>=0.4.34)
Requires-Dist: msgpack (<2.0.0,>=1.0.7)
Requires-Dist: optax (<0.3.0,>=0.2.2)
Requires-Dist: prometheus_client (<0.22.0,>=0.21.0)
Requires-Dist: scipy (==1.13.1)
Requires-Dist: tensorboard (<3.0.0,>=2.18.0)
Requires-Dist: tensorflow-cpu (<3.0.0,>=2.18.0)
Requires-Dist: tensorflow-datasets (<5.0.0,>=4.9.6)
Requires-Dist: tqdm (<5.0.0,>=4.64.1)
Requires-Dist: transformers (>=4.46.0)
Requires-Dist: uvicorn (<0.33.0,>=0.32.0)
Requires-Dist: uvloop (<0.22.0,>=0.21.0)
Requires-Dist: wandb (<0.19.0,>=0.18.5)
Requires-Dist: websocket-client (<2.0.0,>=1.8.0)
Requires-Dist: websockets (<12.0.0,>=11.0.3)
Description-Content-Type: text/markdown
[Description omitted; length: 9881 characters]

WHEEL

Wheel-Version: 1.0
Generator: poetry-core 1.9.1
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
easydel/__init__.py sha256=AyZXMCSWYTaVdWYtQM0LCUkWR113lo7N9ST7r6ESS7Q 8451
easydel/cli/__init__.py sha256=DQgI_xKvkUuYCUS76lSjKvsqlyqVemM8zMTnkmYZUj8 617
easydel/cli/cli.py sha256=EocqXRg0ya-6jJupsMYqD4A-JapSXamzEqwxyp6MhtA 2295
easydel/cli/serve/__init__.py sha256=DQgI_xKvkUuYCUS76lSjKvsqlyqVemM8zMTnkmYZUj8 617
easydel/cli/serve/engine/__init__.py sha256=DQgI_xKvkUuYCUS76lSjKvsqlyqVemM8zMTnkmYZUj8 617
easydel/cli/serve/websocket_engine/__init__.py sha256=DQgI_xKvkUuYCUS76lSjKvsqlyqVemM8zMTnkmYZUj8 617
easydel/cli/train/__init__.py sha256=DQgI_xKvkUuYCUS76lSjKvsqlyqVemM8zMTnkmYZUj8 617
easydel/cli/train/clm.py sha256=kZ1Re_Vo7poi-7lbf5WSdbxtAxkRrQxq0meEzbCC0wU 888
easydel/cli/train/dpo.py sha256=OP2rLL0YoKYY7PAgYuf-TeXlO237UsRo6Ls5EqWC8jc 888
easydel/cli/train/sft.py sha256=CwqDI24Cd12zHcyIsAXrPHaapxdjz70OSDzpuu9IhjI 888
easydel/cli/utils.py sha256=ZrUNKfyoUSZq8HDgPQbPrJl2PVXoYcRjkL2HgTOPJeo 4332
easydel/etils/__init__.py sha256=zXISrH5EnquGNj-3sBIceIEW7PZsi1n3lVBUh2fIXqs 618
easydel/etils/auto_tx.py sha256=Gz4ODCk3XRG5TO0HgAM_KTOXabVt9hOdFGbOXBMkkzw 8928
easydel/etils/easystate.py sha256=9TLzWo_aMVGpJ1LL8KqkyOpmfXoTsbEGEnRJXHaIV9w 40332
easydel/etils/errors.py sha256=ZR_pw5YlcRZ4owfRc3Osg_Ca2rsbIMCJl2AfbI7i7vM 801
easydel/etils/etils.py sha256=NkY-gP4o2lPvdiT4IIER8tsgWaAS2gNSx5wxHVHDrDc 6564
easydel/etils/partition_module.py sha256=wXbPqsSLOBVEw2W82o8ovDfu1bQikTR_oGyA-GCNkLI 3143
easydel/inference/__init__.py sha256=aJh1EORSqTfmyOWEVgp1_TSiXEcsiXls0ef_Tcm4omU 994
easydel/inference/logits_process.py sha256=1QXfJizdwDM_rQbWv0syWeOSUjYb6pgULQrfElCJ6qo 22473
easydel/inference/utils.py sha256=8Lqr1a3j3J_klYNAFbioTutx8pcG1Ji5kS2H1xhlp6c 10400
easydel/inference/vinference/__init__.py sha256=B1OmmbVQHxH6KVxUtWoRjGAYiZknB23g5hAjUA7IqLc 839
easydel/inference/vinference/api_client.py sha256=FAxZZpCWhxHf80Eo-sdIvEOOWvuTInRRjrGMDCthbS0 3611
easydel/inference/vinference/api_id_generator.py sha256=fxl--6KR6tBJ_SbScKL-GO1Kp773IFlOAICkeLuAVLs 910
easydel/inference/vinference/api_models.py sha256=psxKoXrIpRaYeL3yGyfap7kDkHd1qHB05rXLDgN_t7o 2824
easydel/inference/vinference/api_server.py sha256=Mz79EWbyo3EvUF2J_1JfPiUYvMvKRqui71kFWQIsf9A 11812
easydel/inference/vinference/api_server_test.py sha256=2bJR6PdAki2ILbxWIs0usGm5CpR1Gj5vCC1wq5RmO2o 1463
easydel/inference/vinference/metrics.py sha256=rL_ASqb_jwefEqEEw2LGEwYvuVY6hB4---yTPnreWUw 6534
easydel/inference/vinference/vinference.py sha256=3UK8iZgIzh3IKy87htkS82fNpLfyo3MIEGaK0g0tS-4 24163
easydel/inference/whisper_inference.py sha256=43Yg1ulr4tqGfEJ4kkwyf9wAzUUiC-Sq6TOtNB-tzHo 16876
easydel/kernels/__init__.py sha256=INn4U2n2pTy9XB7aaDKtbobjvwQ5TiQO2IcqddSA_fQ 649
easydel/kernels/cpu_ops/__init__.py sha256=Gv0rzIK1Qg_v5Ibx3RVBHOfVPiRlOIAvFlgL4GN5ByE 776
easydel/kernels/cpu_ops/jax_mha_flash_attention_2.py sha256=CNRKGMQZTb9gZE0KqHNdLYnxLz5XBZ1JU84wV2GLbLo 15714
easydel/kernels/cpu_ops/jax_ring_attention.py sha256=CzM_xJzBggJLsJdFutVPWGNLfV4p9ZQV4E9hwEez1Po 26095
easydel/kernels/flash_attention_2.py sha256=fCN8JsZo3JbDQVoSx3QW0NIIQETvmvwNFHzUpgLZQxQ 14031
easydel/kernels/gemm.py sha256=nFI2JyjSbfpJBSXBArlItlMqWsS3GS9AsBK3b3wd_wk 13202
easydel/kernels/gpu_ops/__init__.py sha256=6SfNG4WggssPb8DWBP3gwH3V5Kvro1U30cjyHjZUKAo 1257
easydel/kernels/gpu_ops/pallas_gemm.py sha256=BVHbuhxAAJIRhj0FOaPdjMl_2chYRt4OuzS-Ggrow7w 7348
easydel/kernels/gpu_ops/pallas_gqa_flash_attention_2.py sha256=rm3bTEnxZitHWwTu8SCd2YkuYWMxov9Szq0nDUxu9D8 16419
easydel/kernels/gpu_ops/pallas_mha_flash_attention_2.py sha256=MitXH0n0vdZqln1jfTEHaI92-e0a3sZHXgYAN2Dytps 15077
easydel/kernels/gpu_ops/triton_gemm.py sha256=2849tGaGVBW1pamsOCnAfa7n23hKRJKjZ_MM74qgAK0 12099
easydel/kernels/gpu_ops/triton_gqa_flash_attention_2.py sha256=oZ4_H-hZmvanQ0musEXVRjSNBMOHU53iFkUFI4Qsvfs 26898
easydel/kernels/ring_attention.py sha256=q0jYeZSDFAOzbxBa-yi12HNmQOkINOKOa-M35rhW_qY 8408
easydel/kernels/rms_norm.py sha256=tlxzFiI_n8BM4yvahM5nUVwOhEIpBsUc7lzZ158eJLA 6654
easydel/kernels/tpu_ops/__init__.py sha256=h3y1e7IZTWbo_pfVeTlGNWTNFJ7ZfZ49P7Lasw2ENsM 833
easydel/kernels/tpu_ops/pallas_gemm.py sha256=i3HRzxWhhWLX8A1s9J4JyG7TvWUGP-wJeLsjrIFxMas 5438
easydel/kernels/tpu_ops/pallas_ring_attention.py sha256=Y0-fhff9ZgPB-fslzLcgX1rMGnz07LMVDV-_aMIhe9E 52151
easydel/kernels/utils.py sha256=t3JcwZnGfvwuDJ37AkqH5cRtQZbnj_Gg1ADos0YUn7s 825
easydel/layers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
easydel/layers/_blockwise_attention.py sha256=lPk97Z56c_Q9E9kMa8H7p5qVgXvqsTcYHQZcgCAZB04 6382
easydel/layers/attention.py sha256=WM4cgMABtB9XHtgNXNghq5K-jMnC8kxhSrkaUFFor0U 59607
easydel/layers/common.py sha256=FNm2fAIegYiFJ94fuIxLxS77iePxnLXs6JS85R_Sq0w 1881
easydel/layers/norms.py sha256=pI0Xi67-LZ_ErYGeqD3jcaw9SdqQKCIBBM3LakvlNAA 1834
easydel/layers/rotary_embedding.py sha256=nEfXslJ-xJjJAO2b2xW8sXPg0fNu42RzPIzi4YVLIuA 30501
easydel/layers/rotary_embedding_test.py sha256=Q77eLOzUVDm0OpjhQKgMQWe1DbqxBnNgYjmbY7VQ62Y 6584
easydel/modules/__init__.py sha256=YDRloFbuOs0o-tgyrCwN6zDL-awW7sHR0geueUSc4Fg 4648
easydel/modules/arctic/__init__.py sha256=iI-7908LREmlgm1D4XIASPYq6mFZrT1ivpMleJFDSXY 859
easydel/modules/arctic/arctic_configuration.py sha256=_KNPLbuTotI0CUIq4MZKsSg_Uqfg8pIqXQ_JoGak9dQ 10405
easydel/modules/arctic/modelling_arctic_flax.py sha256=NPxQxtrpGnaC2u5-p3B0ZLcBZEnfquR71TMkBTrAH44 34480
easydel/modules/auto_causal_language_model.py sha256=EFLHdVszCFfv1dZQqJaVgG3hpxVuP3-MsAVl9GsADeI 19320
easydel/modules/auto_configuration.py sha256=NMVC_qZbc1o2JNn3L9oqBC8hfnff_5dSfT7kjJkjdnE 10937
easydel/modules/auto_modeling.py sha256=0CGm_WC2bRF5Doo4BU05blY2n180gupuQOwPJzhMPdQ 4821
easydel/modules/auto_speech_seq2seq_model.py sha256=RBdRCgGq9J01uiDtJPKz0aH5lyoxF0bYiY4bVeiy_bU 19502
easydel/modules/chatglm/__init__.py sha256=vphZinFbxYeA4ONvN6ereqsEz_38On3lzl6jo5m7_4g 796
easydel/modules/chatglm/chatglm_configuration.py sha256=d_rW--ifngbpWZtILoZwS4WiBqte8lx2BVb-kMMNoVY 8571
easydel/modules/chatglm/modelling_chatglm_flax.py sha256=CehQeAA-o8d-miU147kbejpQ8zQcNugbeNU5ahpKhBQ 44103
easydel/modules/cohere/__init__.py sha256=wLTzXXRTJoT9MElObuPlobDFyRSssWhlBC0noMZmCow 859
easydel/modules/cohere/cohere_configuration.py sha256=tt-fESreUiLyLKP0vDnQtYFy7tnAQ7y9l1jLPlf6kY0 8074
easydel/modules/cohere/modelling_cohere_flax.py sha256=J-oNSZoXqhlSRv5p8HjFMCM7AF2vSAs-P7NQs35NHYw 25552
easydel/modules/dbrx/__init__.py sha256=PRietVkQWm1GCzA4eKgri6bcJ0V-PTmfUdo7kZ7oMRc 933
easydel/modules/dbrx/dbrx_configuration.py sha256=d8MV2fk7BKTdfGnzoKxfEGZ6gSVkscsP3LnZ5K56GJA 9899
easydel/modules/dbrx/modelling_dbrx_flax.py sha256=0MEcFmgpeP3zQ5ZV0w-ccCZFb2TVyrkB1DNUx6QbjOg 30903
easydel/modules/deepseek_v2/__init__.py sha256=XB4Qyh9t9mzkg__KFPd1oCKFMqRFYxELCCV33Apq0vc 896
easydel/modules/deepseek_v2/deepseek_configuration.py sha256=Eu5Us6cAJhJfL0zM04GxyGlKnGL2gFBa2cK9F-HzTDc 11895
easydel/modules/deepseek_v2/modeling_deepseek_flax.py sha256=59LaBGOKbUEoDYKWAA6svQNcgQek-I3xnLGMNzPqYg0 35932
easydel/modules/exaone/__init__.py sha256=D_m0tLhMHYs9ei1nuoKa36MZd9Pp-tes6v6rF1FpEG8 858
easydel/modules/exaone/exaone_configuration.py sha256=USatqKaBavA_dIEvTsNNGvllgvjUSpNeg1k27H32YBU 9323
easydel/modules/exaone/modeling_exaone_flax.py sha256=AZc-eNpohbxX2njx3ICa4fdm5HziomUous0aNwhO3bo 24409
easydel/modules/factory.py sha256=mrqe9BzAFHNIcauapD93jLBr5cTLdL7l1r60lkfZWGk 3710
easydel/modules/falcon/__init__.py sha256=aXwmmlq22Kig_PuHp6uwS7GNs0oXZ1x99sXthnPweeU 858
easydel/modules/falcon/falcon_configuration.py sha256=zTkhM0uWv_pYxVvmZkbwz0QWXLDr_VRyAsDccYm6WAw 8946
easydel/modules/falcon/modeling_falcon_flax.py sha256=ckgTVhIu6E-cwChIGotQW92zWKlVixuGjvb_kcmUTqg 28875
easydel/modules/flax_modeling_utils.py sha256=nkZxPjI_4XGK3okeFcgvqUZfca5XlZIOXA_PoQpkux0 37560
easydel/modules/gemma/__init__.py sha256=lXoUuvaIThuMlF9fbFWVsrFwu5ojjI1sW57vLmc3CnE 848
easydel/modules/gemma/gemma_configuration.py sha256=68GQ6sTs_VXXl2WZXMOllmT8iKNWiN0wS6sI7hCc9ys 7995
easydel/modules/gemma/modeling_gemma_flax.py sha256=pDbcPR01s81w0MJ3v07lp61WCTKTX6dT5_KCC-M3mjg 22548
easydel/modules/gemma2/__init__.py sha256=DM15uGmS65eNwJAdPoN2EDOn80dAj-PR9RVyJ1YBrA4 858
easydel/modules/gemma2/gemma2_configuration.py sha256=LnAME8NCae3k1DH-cSvOebZ8Jhm4j1KeCapYRtqYTa4 8816
easydel/modules/gemma2/modeling_gemma2_flax.py sha256=_1s6kp-pCsjLQ7MGnqSIy3R0W6PgJopeFOJJpjfkPxE 26129
easydel/modules/gpt2/__init__.py sha256=yY0yYrb1UYz38kpgZSzVl0myUZ52FSnGfpGZQ1X81JM 838
easydel/modules/gpt2/gpt2_configuration.py sha256=mCiOjC_DRjxgNIjTm1mn7br-JDnEJQg7WhAUhcnL1SA 7961
easydel/modules/gpt2/modeling_gpt2_flax.py sha256=K9osKjGEI8QgRo-GFtEM1bCEKpq-xh3LU1yD7lWxGSE 19817
easydel/modules/gpt_j/__init__.py sha256=joruadUFL_w2atAvqUTNsC9NHq-L8UavEDJ5-E_7xSU 842
easydel/modules/gpt_j/gpt_j_configuration.py sha256=Zm3WoKuepPA79UTAwxPArA8htHRSMn0PN0CpDo6HN1A 7644
easydel/modules/gpt_j/modeling_gpt_j_flax.py sha256=44Tzu_1Dd5XyrmhhaS5pFhPspm4zdIcuw4qx2nKJHWY 14201
easydel/modules/gpt_neo_x/__init__.py sha256=Hh_NAvxZnJYRXxOWAzzmdMN7wLxfya7Ki0qZNlmzbMY 876
easydel/modules/gpt_neo_x/gpt_neo_x_configuration.py sha256=ZQq8cP63B-xOieyvw_8W9IqWkFfVRhrsF8zkO_4k0kE 6387
easydel/modules/gpt_neo_x/modeling_gpt_neo_x_flax.py sha256=MEM8rovHGf9rAlpgYmPuSCkN1GNU-BUr0ZBBkqtSn_g 11208
easydel/modules/grok_1/__init__.py sha256=4G2TnhTp4HPCq2Jzh58wUHXEKpvclSejzldQwa4QccA 852
easydel/modules/grok_1/grok_1_configuration.py sha256=CrIU60dsBt8ldNZBuvutvO1WFpEnDx9x6tarM5vkyz0 8277
easydel/modules/grok_1/modeling_grok_1_flax.py sha256=aoQ_q-7N9lIOQja6rxFomrv1E7FAaJ2VdIS1p9nj85c 31150
easydel/modules/internlm2/__init__.py sha256=rQNLr8Uud5dBucnmpF9VQo4uXQ0GMRwctdtA1ZohW8Y 980
easydel/modules/internlm2/internlm2_configuration.py sha256=pLNu1LwjRxGKCwFbJ-WTpV6avJj9c0xuPeHb1_uMS5k 9474
easydel/modules/internlm2/modeling_internlm2_flax.py sha256=OyRvUJcZPKP58y1I8RngXND03vKGUoHcWgnGyZ6KUOE 26639
easydel/modules/llama/__init__.py sha256=7ZsDEIqjpASHCa-PPigIdHSVNqWKVB9bZPygXotNEFo 889
easydel/modules/llama/llama_configuration.py sha256=ooHNdAxBEhrFbUo9eru0cXtcDXZtjULnUYOZ-nm2jAM 12760
easydel/modules/llama/modeling_llama_flax.py sha256=RnoyyrnGYg7vWZOp_S9G9ZtRZxmQRL-vN9fM0wx-Vqg 33404
easydel/modules/mamba/__init__.py sha256=QlSMOuzaikVfwfmsF3-1WEaKssF5ABzYXX6WCEEkMk0 991
easydel/modules/mamba/mamba_configuration.py sha256=yVV544wWXwwaXPrTsKlymYBWPUGRGGTA11fyb07yQ5M 6583
easydel/modules/mamba/modelling_mamba_flax.py sha256=clojnUoRKKO3rOaesdMJ_CLKg-TpFVzTFtqfTWFSwjg 30766
easydel/modules/mamba2/__init__.py sha256=PGZEx8jIN2lkVC2XXIVPIbltWf0kH9ZsoZL0WYV7E4E 1007
easydel/modules/mamba2/mamba2_configuration.py sha256=QLVjQ288wh8jt6Lw7Wop7oo0E4tLEbyDlKrSui47KVM 6666
easydel/modules/mamba2/modelling_mamba2_flax.py sha256=WGj_X3L_n1R8tas7FiOkGXxVJyf4OcGNDfl-TntEdqM 39585
easydel/modules/mistral/__init__.py sha256=DXZeKfRYBeV1SeCDSnOLCs1XGV79Z5L5CmOOwsb2PIA 993
easydel/modules/mistral/mistral_configuration.py sha256=LdsRRyhvVZqlncOBHWRFQyGAYfHLnld61GU5RPv6Zys 11146
easydel/modules/mistral/modelling_mistral_flax.py sha256=XUB98EvmHh_pht6zFvA3sa5L-nAX0w0qKujajKxn-0o 32365
easydel/modules/mixtral/__init__.py sha256=ynOoWaLPEImVPGGazqXZ0xXahNUPZv6bZPk4sTGcHAY 869
easydel/modules/mixtral/mixtral_configuration.py sha256=gGqwso02qHYiuDMRotrhtsXZJMK_KcHUKZAGpqBecwU 10979
easydel/modules/mixtral/modelling_mixtral_flax.py sha256=T5JL_jojCtOhR9aCMPZkvPxYhm0kfuuceZ-GBatLgk8 26481
easydel/modules/modeling_flax_outputs.py sha256=PiJz5BW9aPf0Mrr3dGT1I5rHNQ0563tpr0hyoyR2Euo 47139
easydel/modules/modeling_utils.py sha256=9P4gKObjY10ukMfh0vdy-1-ykvjGjij4U5GS2Pld7Us 90027
easydel/modules/mosaic_mpt/__init__.py sha256=Wt7LJzLxJ--mgOJrTJBD3xGENjNKHIBfxa4yNuj1OEE 904
easydel/modules/mosaic_mpt/modelling_mpt_flax.py sha256=oUHGOdyKsRMEvIAsKJJucgXpvgQKnIhBd74OkNw2Rfc 21453
easydel/modules/mosaic_mpt/mosaic_configuration.py sha256=IHFJVezbfl8hwaeIxPcamGUX5PdUsVsCaVznQz_PXOw 11185
easydel/modules/olmo/__init__.py sha256=JFHijv9kmb69DU8JgSqHzkDo8sOyZs5WoATUjQi6_jw 838
easydel/modules/olmo/modeling_olmo_flax.py sha256=o-NNmcPHJPpZjASxk5TxRCwxW5aKvSQY4fIHve3hhFk 22591
easydel/modules/olmo/olmo_configuration.py sha256=3SaCd2PyAyupiyHD95FdJsdyYviOKIczwE6OFEGMs90 7768
easydel/modules/olmo2/__init__.py sha256=jkWchhO3XLfYA4Ij9_wNC66kqM1n-EHvjLnjcIX8gbs 848
easydel/modules/olmo2/modeling_olmo2_flax.py sha256=8nzUbqV7Dmd9FyE_hHSJK0Hf8DWNvnP74oicd-78ywk 23101
easydel/modules/olmo2/olmo2_configuration.py sha256=2rUSEKDT6FVGBF6uwL9gzPlQSku7PS0E1-G5-sh1sYg 9707
easydel/modules/openelm/__init__.py sha256=Es001cCOr9D0__E4Gk98JILOitKVM8lsWp6S4FSlamI 869
easydel/modules/openelm/modelling_openelm_flax.py sha256=iZC5ySUh4xeKbFU_RrF7uDel9ioK7ZWnUMh3REaHJVo 25617
easydel/modules/openelm/openelm_configuration.py sha256=60r2X7SPjqObDs7TGN_nZOyiYdwCQF3nQJWGV7a8tlk 16654
easydel/modules/opt/__init__.py sha256=yh25NnZMCUo-ipT6eFCU3ldXcxYO6MTOV3ubtTbXwSA 829
easydel/modules/opt/modelling_opt_flax.py sha256=QNutxVP_nkNyoVXgd63Ed4XIciKz7gT6B-y6KCrfDSE 17327
easydel/modules/opt/opt_configuration.py sha256=NtbpPBLvr2uDem-oWg7y9K5jFmoJZ_-T2nUC6RAqDKI 8262
easydel/modules/palm/__init__.py sha256=te9eie7lBKRlSBtFsGK2V5KWLG_Jsg0roITYT2FuOrw 839
easydel/modules/palm/modelling_palm_flax.py sha256=xjUTRlrZXuCCNS6OQ7gOZE2mv1lllVw2vTbG3jtmFAE 9655
easydel/modules/palm/palm_configuration.py sha256=Lunh_aGAGYkNfXnyfiR7AvN2lAdIqmmNWwDO0Mu_yRA 3028
easydel/modules/phi/__init__.py sha256=nPsZiTo0pNBtnuQ7ll4k5G9-fAN4SBoHwjET10YW5vk 829
easydel/modules/phi/modelling_phi_flax.py sha256=a2drJ4whGoFL7GomoA-xVNUkQuKRgFyIM7HeC3d6oio 24533
easydel/modules/phi/phi_configuration.py sha256=_QazJiZJ879CVffBqu43qvui31VLIKia7kruwdi1_co 9625
easydel/modules/phi3/__init__.py sha256=rAvM-COvh5VgVHlXFUPIY94I_y95KAeXqvV0gcH4NTE 839
easydel/modules/phi3/modelling_phi3_flax.py sha256=Jqu8KAuAL3hs0n-Bz_oOK5glP9QBrehAN2ld2lO2WTE 24349
easydel/modules/phi3/phi3_configuration.py sha256=UrgnQxsA0UFEWc9QqBjU1icXE3vJsZUFn42-gYJkcpk 10999
easydel/modules/phimoe/__init__.py sha256=3JiGTX5bAlydK-ThxPNMVuaWkTDzNO7blnddB7tCB2o 858
easydel/modules/phimoe/modeling_phimoe_flax.py sha256=eFT639HpaACux1rQZQv-MycssJZXnyt4-e836PLo20E 28412
easydel/modules/phimoe/phimoe_configuration.py sha256=2zd4Ag28gTE_lXiZ3yArSGhxqj7gt5ecNVLmbHivpBk 13469
easydel/modules/qwen2/__init__.py sha256=PWuw5rcP-HDSyavRDCwIJo0C-qvdlyjuY-kuNHVwwa8 819
easydel/modules/qwen2/modeling_qwen_flax.py sha256=XUdHPWxi-6Y18uQtqDC0WBwZScGYZ57IclpOTJjAjlo 28802
easydel/modules/qwen2/qwen_configuration.py sha256=_-bask4GXJcwzOF3gYWYcJF7Dqx0sRXoZU915KZggAY 11530
easydel/modules/qwen2_moe/__init__.py sha256=69qyEnovGJwf3xRSwfnyuhNEiLTknvr-XYIPXZZPWxQ 882
easydel/modules/qwen2_moe/configuration_qwen2_moe.py sha256=DDEwHnBiCRQYLEHDIfeQnvBjeWkBwjj8upLWAY7-Ncc 10242
easydel/modules/qwen2_moe/modeling_qwen2_moe_flax.py sha256=tplDazS9E3Xp2fsOgIOFq6xOl1cvdNxq13Ze8TZYhz8 34601
easydel/modules/roberta/__init__.py sha256=MMo-81iRSQoa35DRIff_UIpF9dibpPfDGVTFYecVmt0 1125
easydel/modules/roberta/modelling_roberta_flax.py sha256=98CqN-mg1D3-qWhmopxbSi7qGij9_LvSh6_dVji_l64 36753
easydel/modules/roberta/roberta_configuration.py sha256=GYSoIaJ3JyLXOcJ0LfjUkMcjNU1KSoElBk-uzpPd0QI 8411
easydel/modules/rwkv/__init__.py sha256=B9isy9YK8exdBVdeYvxoDZA19c546RScs-6kQttkVoA 839
easydel/modules/rwkv/modelling_rwkv_flax.py sha256=dKpEChsF1t7VEnjitevLv4N3NvSwV0vcxrpPA6uEH0U 22057
easydel/modules/rwkv/rwkv_configuration.py sha256=rHrbN5WT12fsRPfAjo0rwAf1p6l28LTZR-dtOxzmoZA 5386
easydel/modules/stablelm/__init__.py sha256=TVPdbHrI-8WUDGob_R5CeZe_0z9uhQNuNtINjqKt4S8 879
easydel/modules/stablelm/modelling_stablelm_flax.py sha256=1BloN6-2-VEOukVRve7caNtnCC9qsPJ3Msyman7dxJo 28519
easydel/modules/stablelm/stablelm_configuration.py sha256=ftXIkVC1KyHivFyChIux_AxAM6pQRQPdcccG9E_PIVA 8869
easydel/modules/t5/__init__.py sha256=a7aOz_zr_e4VbxlmpsjtAIq4Wd0IYoksGN1gUZV33tU 845
easydel/modules/t5/modelling_t5_flax.py sha256=Bjs2aHUV7j2CEt1OyenBn25hzrqKI3kLjjd5Xm4PpJs 46798
easydel/modules/t5/t5_configuration.py sha256=327_Wgs8UnGLFrpntpm2_tqkU_hl4FUTGMloH9g5ySc 6357
easydel/modules/whisper/__init__.py sha256=YkSspD1pMlwzPi8BT7wQ7hPkyZNEmWz5JL67GL4nNAA 950
easydel/modules/whisper/modelling_whisper_flax.py sha256=vBz8DnOsmOCyGiKyOaBE6Z08xE3kDm_INJaZ-CNc98U 52440
easydel/modules/whisper/whisper_configuration.py sha256=7aujQ1iYjyLRgF1782AgnlY2IWZrUHthDEsxxPjTqsA 9939
easydel/modules/xerxes/__init__.py sha256=GlwJ4laJxdSbn70Mujf5X8sbsovMl4PibYJ7ieWboKo 858
easydel/modules/xerxes/modeling_xerxes_flax.py sha256=0_wpk-LT-n-i8AhnCMQ33sHt0NjLpx8E_b2wXMBPc-g 26098
easydel/modules/xerxes/xerxes_configuration.py sha256=QnMPP5242brtF1PmRGmvrp3-SaICEQXaN6YUa-_odFQ 7753
easydel/modules/zamba2/__init__.py sha256=DQgI_xKvkUuYCUS76lSjKvsqlyqVemM8zMTnkmYZUj8 617
easydel/modules/zamba2/modeling_zamba2_flax.py sha256=3QHulRc8UZzEQ-8Mo3KM5TrXUTjMN4Zai5u-Xqkt--4 10371
easydel/modules/zamba2/zamba2_configuration.py sha256=djKcqegPw4Uxhtq3-75E5c636Ov4Z-xVhvvmH6rk1tc 8828
easydel/smi/__init__.py sha256=okWYH4d0yzqn-ECUKrk-zG8V9mh7Dv-Z7mhZEpN1eOQ 700
easydel/smi/smi.py sha256=p93L72Z_0Jc6--0_gJlsAvnB8yu3WUGx8bYhYT3scjc 4532
easydel/trainers/__init__.py sha256=hE5_bdTbBlU7yTEnekZdjR4rZH3V-WqGB0I2i4R91dg 2513
easydel/trainers/base_trainer.py sha256=oqPIiYaHFiWHlqO5WzXkThuc0wnEhMd19qs0FVyLPGM 41646
easydel/trainers/causal_language_model_trainer/__init__.py sha256=85T29Q7aggborPo5XwfY_-bgNg0Gwm9D9uqxccHkKys 763
easydel/trainers/causal_language_model_trainer/causal_language_model_trainer.py sha256=sgfnMtKY60uZhvo50Sjfs6SWo4WHXKzWGm8W7wwUuPM 33731
easydel/trainers/causal_language_model_trainer/functions.py sha256=NLZrdDz4iBcs19uZkqCnZrCba6VhviaW2FHOGFL9Oxc 7400
easydel/trainers/causal_language_model_trainer/modeling_output.py sha256=RiBuXq5CkYIwAdiG1Q_36kGwDtiywKbHDGeAT9n3QcM 1121
easydel/trainers/direct_preference_optimization_trainer/__init__.py sha256=meYXntpJ1MZGcyrd95ghg3AjRiVYarGcDFf0cD_Tqbo 809
easydel/trainers/direct_preference_optimization_trainer/dpo_config.py sha256=oa0ju6I1tJJZXvhigrrdVzn_C4nCDii41BZuwva4pQ4 1839
easydel/trainers/direct_preference_optimization_trainer/dpo_trainer.py sha256=Nu47gg-WXd83X9aZvODeLzD3XvUGEZFLS0st4TOpgVk 42863
easydel/trainers/direct_preference_optimization_trainer/func_utils/__init__.py sha256=rokLRNvxUAbA7_-sIh3yp3NufXIab4dy5v5kMLnpSM8 291
easydel/trainers/direct_preference_optimization_trainer/func_utils/concatenators.py sha256=a0pacGuNHHZ8W2G_XOlcpmRcujlSslF5SPO16T7Expk 4274
easydel/trainers/direct_preference_optimization_trainer/func_utils/creators.py sha256=zojljKsycmVlH5HGq8Jif14aTSdTRNRgLtqXZgrq_9g 12296
easydel/trainers/direct_preference_optimization_trainer/func_utils/log_probs.py sha256=O9sTHzdwmPJyG-MAjpmLhBqnMTTlnWgJTcoLeVn8-uI 2343
easydel/trainers/direct_preference_optimization_trainer/func_utils/loss_funcs.py sha256=cUfI_CKQH571wfmaRQ8-crlnBgPlmUFjp1TB2vNlVFo 8751
easydel/trainers/direct_preference_optimization_trainer/modelling_output.py sha256=eBEGGRFbMT1zNVQlncQeHE83iSwkXJXy9z0_sVi3jAM 1116
easydel/trainers/direct_preference_optimization_trainer/utils.py sha256=8v09mt5fVIfDu5xvrVraoQWvXdiKDgOQSd8n92GJ8s4 16898
easydel/trainers/odds_ratio_preference_optimization_trainer/__init__.py sha256=H1EZ27vrHWXfhDnuxLXLIXJLsEP52EAgHe5S7fZVzHE 753
easydel/trainers/odds_ratio_preference_optimization_trainer/fwd_bwd_functions.py sha256=GKZLbda5rzAKOuIuqyrR3ZTn9x8wG0rSXCHPDs-hDMs 14906
easydel/trainers/odds_ratio_preference_optimization_trainer/modelling_output.py sha256=Y__r39uE1NqKVmdUfSkvBrbfetEOiH2m7PXCB7kyZc0 1127
easydel/trainers/odds_ratio_preference_optimization_trainer/orpo_config.py sha256=LQZEqFIVwvlHqr8q2yzVI9KhK9QZpAbuM3BB-j5rkXE 1502
easydel/trainers/odds_ratio_preference_optimization_trainer/orpo_trainer.py sha256=EMhqS15URjk4E4oWwIEzWyqdsQGLRYuVnRSjIVRXSeQ 43008
easydel/trainers/packer.py sha256=zhoByXiNNH6CIHtCjzol9ELODUlhLw1s2wTC8YwhH4M 4270
easydel/trainers/prompt_utils.py sha256=bVwWSa5r7BNnZygR_9OKWPTrgEJEqlex5f_o7B6MKQU 14594
easydel/trainers/sequence_classification_trainer/__init__.py sha256=RoDzN90Y1ZvKhZGUDYl-L_u1Cuu7mG6ITVlf8LPkuMo 784
easydel/trainers/sequence_classification_trainer/functions.py sha256=d324A58O9Yw6UmthUDMjIU73yhjIhqDvsWgy6bX4umo 8422
easydel/trainers/sequence_classification_trainer/modeling_output.py sha256=dEdzK8FCvGBSQZJ8cT6KDxYN4R-6StADlNou1OOEh04 1135
easydel/trainers/sequence_classification_trainer/sequence_classification_trainer.py sha256=jU7RMHFIUB2GZbh-vY6aMcV4Anav9e3Pq2YudAvW2Po 25303
easydel/trainers/sequence_to_sequence_trainer/__init__.py sha256=EQg4nKCCNZkQgLg1u8Hm8Trwd2ip9D7v7vRIUswqc-I 292
easydel/trainers/sequence_to_sequence_trainer/datacollectors.py sha256=KW0V4OZSDiPcn3yssdKur8LTvABz7NlzCsg3K7rNFkA 2165
easydel/trainers/sequence_to_sequence_trainer/functions.py sha256=eVSwwqao0SzvfeOAiB-t5E-3mSQHsGYBNJPHZqeIikQ 5902
easydel/trainers/sequence_to_sequence_trainer/modeling_output.py sha256=UoJ6it4cbGcktn4KgpP3zhnfOH8LMJcFWgf_xFnrBLg 1120
easydel/trainers/sequence_to_sequence_trainer/sequence_to_sequence_trainer.py sha256=6ys9Wgio2YOhNOBe0J_fGBtR-EB4mwSdrnv0-_AsDPU 24919
easydel/trainers/supervised_fine_tuning_trainer/__init__.py sha256=vD-fZh0jslTywozugs_1YYQqHZYThqcerzb0Td8oeQ8 700
easydel/trainers/supervised_fine_tuning_trainer/stf_trainer.py sha256=CaY4xL7gmBlIqWW-fgdWsU3vzAQDU4ysQiMf0BU7qr0 21754
easydel/trainers/supervised_fine_tuning_trainer/utils.py sha256=DQgI_xKvkUuYCUS76lSjKvsqlyqVemM8zMTnkmYZUj8 617
easydel/trainers/training_configurations.py sha256=-xJyn_Nc4xvJYYaYw952PUBaJohFIYYY8ZYB1Id_Jh8 25860
easydel/trainers/utils.py sha256=mORh31jbQN6QvCCGlLgRmsQaGFsvENmDk0u1MV0HtYE 24404
easydel/trainers/vision_causal_language_model_trainer/__init__.py sha256=o8dEYa-HTgv8JJ1uD_lVpJRhOBUTbcFrXLFAVNyS6D0 789
easydel/trainers/vision_causal_language_model_trainer/functions.py sha256=OiXoYeYHiejV1s4oNLvXog_OTrLpSlERlBzBI_x__W0 6422
easydel/trainers/vision_causal_language_model_trainer/modelling_output.py sha256=YrGm2bB3m_Okg-TLGjoKuA2d8B7RNPFACb0L9LSYj2U 1127
easydel/trainers/vision_causal_language_model_trainer/vision_causal_language_model_trainer.py sha256=zw0h_iBQIWzIySJRXvJ3lAdMDwDDrRxS7ha3CleSz24 23434
easydel/transform/__init__.py sha256=SeIZjxi5qzudEfbQ1xe0KpaNGbYWOtjuBWwD93GK-kw 762
easydel/transform/parameters_transformation.py sha256=AobDeLo0Z2j_UaLoKbR785K6wMedvJ7OopOTmG4tjb8 10890
easydel/transform/utils.py sha256=KlrE78y-z1LIzGolzInpPRoe4fete2crbZO3OJ8-k98 3968
easydel/utils/__init__.py sha256=jT6EtdelByUjI5yHSpgq2ENvkTP9gNRSwm0erZ9u4O0 749
easydel/utils/analyze_memory.py sha256=ndv12FJPp-RCKL5-e92yUjG-QmuCnnXNe7ysqT7HFew 10914
easydel/utils/compiling_utils.py sha256=E9KZ8l4PO7gaDn8YMMXDu9zbd2yAHCChmgc0IndRkuA 8430
easydel/utils/helpers.py sha256=3MGW15AcgjWDm1wqlBra6I8eK3gEXtNTYEZYnqIg9fA 4866
easydel/utils/lazy_import.py sha256=1LNvAIpXrH09jOPQYSmPzeW3U6BidlQmFIUnO8sADGE 3011
easydel/utils/quantizers.py sha256=rG3kMbZKQd_70FqrfduT4c4cseqW2WEMIW8Thiyg5Gw 2200
easydel/utils/readme_generator.py sha256=ZhHMx8cPjPc4TzWoh_ZfP6YMgTIaE3jpFUR6Rgl5KC4 2408
easydel/utils/readme_template.jinja sha256=dk8usgCTPdsM-woFYta4y_me4DRcaubNwSUUtKsqatw 2974
easydel-0.0.80.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
easydel-0.0.80.dist-info/METADATA sha256=I4nXViOijQXagnJCeSoOBaJBsAViIycF5on00TThALg 11668
easydel-0.0.80.dist-info/WHEEL sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs 88
easydel-0.0.80.dist-info/RECORD