Wheelodex » Projects » arctic-training

arctic-training

View on PyPI — Reverse Dependencies (0)

0.0.1 arctic_training-0.0.1-py2.py3-none-any.whl

Wheel Details

Project: arctic-training
Version: 0.0.1
Filename: arctic_training-0.0.1-py2.py3-none-any.whl
Download: [link]
Size: 86705
MD5: d6fc163f21b89b09f4d06d6bfb7ada8c
SHA256: 4490ce225313b5c8a7549b2e34b002fcb1cabb6f231ebf429dc951ef511ff03e
Uploaded: 2025-01-16 18:36:23 +0000

dist-info

METADATA · WHEEL · RECORD · entry_points.txt

METADATA

Metadata-Version: 2.4
Name: arctic_training
Version: 0.0.1
Summary: Snowflake LLM training library
Author-Email: michael.wyatt[at]snowflake.com, jeff.rasley[at]snowflake.com, zhewei.yao[at]snowflake.com, samyam.rajbhandari[at]snowflake.com, aurick.qiao[at]snowflake.com, canwen.xu[at]snowflake.com
Keywords: llm,training
Requires-Dist: datasets
Requires-Dist: deepspeed
Requires-Dist: devtools
Requires-Dist: jsonlines
Requires-Dist: loguru
Requires-Dist: openai (>=1.48.0)
Requires-Dist: peft
Requires-Dist: pydantic (>=2.0)
Requires-Dist: tabulate
Requires-Dist: torch
Requires-Dist: tqdm
Requires-Dist: transformers (==4.47.0)
Requires-Dist: snowflake-connector-python (==3.12.3); extra == "cortex"
Requires-Dist: autodoc-pydantic (>=2.0.0); extra == "dev"
Requires-Dist: black (>=23.1); extra == "dev"
Requires-Dist: flake8 (>=5.0); extra == "dev"
Requires-Dist: mypy (>=1.1); extra == "dev"
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: pytest (>=7.0); extra == "dev"
Requires-Dist: sphinx-autodoc-typehints; extra == "dev"
Requires-Dist: sphinx-copybutton; extra == "dev"
Requires-Dist: sphinx-rtd-theme; extra == "dev"
Requires-Dist: sphinx (>=8.1.3); extra == "dev"
Requires-Dist: vllm (==0.6.2); extra == "vllm"
Provides-Extra: cortex
Provides-Extra: dev
Provides-Extra: vllm
Description-Content-Type: text/markdown
License-Expression: Apache-2.0
License-File: LICENSE
[Description omitted; length: 3673 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.27.0
Root-Is-Purelib: true
Tag: py2-none-any
Tag: py3-none-any

RECORD

Path Digest Size
arctic_training/__init__.py sha256=rW4IdmoB3xMK53KNcqrBAq5SshI8QJ9fJcvr7zRnIxc 1955
arctic_training/cli.py sha256=r1f1to7bv3jZWPeg3SInk-RQqohR2RcBTDtS64Jj7Y8 1916
arctic_training/logging.py sha256=bIMxT1VU-4aI8qqWOf4XKK5AblLdCqBMu9uWpc5olLc 2733
arctic_training/utils.py sha256=OcNzkeYEhwZrGacaPPxEONmQGdouh0GEWASfKaOS9_E 771
arctic_training/callback/__init__.py sha256=wHBlkPJC5z4iiFsTNYF6njV8M2PfvpQwpxczOE_mCMU 677
arctic_training/callback/callback.py sha256=QHn5ekSEJ9TKv_en2n7-nO89iWVHnOrP2TnHnBHxFJY 6985
arctic_training/callback/default.py sha256=aL2HlIWo6hxe9vHpiycnDOXuRyfFhqFUYVVYXhKIQc8 1062
arctic_training/callback/logging.py sha256=2puZ83czMk5GvCVasGLqaFT-fTWJtKrAovoohi0apLg 2049
arctic_training/callback/mixin.py sha256=MN_3zDnnT0WJsq2bwDYHC_AkWlPYixzkBkLRt7wRttc 5962
arctic_training/checkpoint/__init__.py sha256=Wzo1rWQONF3hTX701CERAzyj7C38yRz2bEtCYv-Ouh8 734
arctic_training/checkpoint/ds_engine.py sha256=uAaTROKMAlqOfnJqdsdP9CBorbBWx18iq7zA_L4IsvY 2284
arctic_training/checkpoint/engine.py sha256=JNTo0M17Et8BtGd-1_cwfhN-Aq84sAOT41ZzgZNn6bk 3752
arctic_training/checkpoint/hf_engine.py sha256=qKRcdlLvg9J-I3Kj6-UY3EEo0qh97oKSHJwF53WN15w 5129
arctic_training/config/__init__.py sha256=HrUUX8a0p71fIwTknJFMAz7B6kCdspsURYwTVkzOnzI 895
arctic_training/config/base.py sha256=8BegRuY79rcHZE_lDhu1m_xL1bbLTCKP48ZdOuET0dg 1108
arctic_training/config/checkpoint.py sha256=zaFx7ib-V0SO4MZ6L2DWm8pQPSwInGwVUkDWioKtE7A 1838
arctic_training/config/data.py sha256=oNo1TWj3woD9BceUZwvIG23tKM9_aCJejw2XdYY-n0c 3994
arctic_training/config/enums.py sha256=nlArQEB2K6Rbx7W8EFNzZBBPTobhag3YYDMRufgthoA 1774
arctic_training/config/logger.py sha256=tzGcyXyrsnv4Vbv4i_k1mc3MfCyU5zcDsFnYu4k6vn4 2131
arctic_training/config/model.py sha256=vT-UPUqKIkaybykATiFjieiAWYT8jn4RpklYAshDpwU 2594
arctic_training/config/optimizer.py sha256=aUFiz-FfDbnxQ7HKRxnTGqqJnxnwG02oiaxx3OOjV0A 1637
arctic_training/config/scheduler.py sha256=Ra-iSkYh4XHyRFJCZj2Mj2pGMAoX_E4Qwtrs1FGolGk 1446
arctic_training/config/tokenizer.py sha256=ilCSB7XMs3pq1eofp5egIL-sFY9pTRt5cbsFagWHldU 1384
arctic_training/config/trainer.py sha256=szLJBz9fEOv0jZF0gNhvHwfldJZBQBgC5hinT7CP6Iw 11356
arctic_training/config/wandb.py sha256=S36L35OmY7zP-8EKTEPzXj3S_tcM9_228DMKOZQ0kC4 841
arctic_training/data/__init__.py sha256=rlwZ916HyyhtxK9TbUqYjyxG9C8DOsmThCrchUNrTHo 742
arctic_training/data/factory.py sha256=dFMUewWr-Uay17wN_x1uTkHBLa-PLk0azZjvT9ata_U 8684
arctic_training/data/sft_factory.py sha256=21w9yIJ7RzhvUbaH9VuzD89EselMtMKAgizkXfCqdDw 12780
arctic_training/data/sft_source.py sha256=1DmmiA8p-XIARh45qNDo8BvG5TJ8iVdp9l5mhzQce68 5171
arctic_training/data/source.py sha256=PDo2eWi0JQKWGAaSIf6TIS7u-1IPwfm0_OdkHpOtOb8 1477
arctic_training/model/__init__.py sha256=nCfZBRoEg1lyEvruSfXlAojQa8J65YaXyWxhKht6HRU 731
arctic_training/model/factory.py sha256=ezIdXuXmRGY39i7Te2jszqOnkTA6dKCov9IVN5Vkcuc 2768
arctic_training/model/hf_factory.py sha256=NEHGrx_x3M7tTJfJ5v2EKbbCKi138VyAE5N3bAhCzZc 2328
arctic_training/model/liger_factory.py sha256=UEKPMbkMe-RBEn66nQm83qdSYuq2I2ZBaVYMARRCf9M 1381
arctic_training/optimizer/__init__.py sha256=QTP9suwlBZRAcG_6TfndZM_2re2-hpHRN-eP5035itc 703
arctic_training/optimizer/adam_factory.py sha256=34dGwqb9K26Vfv5ePoVt8nSsyiWBFZCDTu7qUDITRg8 2795
arctic_training/optimizer/factory.py sha256=C9hgnpgLUnUBa9JyPUM_zjBCXnU03-QYvN7bP_zfwl4 2624
arctic_training/registry/__init__.py sha256=LL3GqV6DrBp8bzBjJ67yhY5QNfzGVYU9jA8P8llgph0 2469
arctic_training/registry/checkpoint.py sha256=38u8vMpcWIXYj4U4T4vd9xP8E5CMzjt26dke5xC9Ns8 2636
arctic_training/registry/data.py sha256=XuD9kVgFKYpxh5s-DjCP-K9EYuJHZNiPPmIM_SIr1II 4745
arctic_training/registry/model.py sha256=mVee8mOjhqh5a89txo_J2prseoTn31bpooBjdUbCEak 2808
arctic_training/registry/optimizer.py sha256=Ap5DAj16d0vOe4QYMpaXsb42gALS4MSjWiuC4w7n5WE 2865
arctic_training/registry/scheduler.py sha256=BRo-3b4G7Kh7NYtGSoxMTOzmLackv2irnya-1lj8DOM 2834
arctic_training/registry/tokenizer.py sha256=BxpSsAXoIU1ORFUPqRtII04taArotg6wugQLwaYsR2M 2820
arctic_training/registry/trainer.py sha256=jhbuPrsQHcFMXzxmYGY8uMobFHxlNYQycG4M-yiJ4ms 4877
arctic_training/registry/utils.py sha256=2EvsvBQJNvzkVhauPT1gFj1UGzxZ7s6pDApKm_Ao_mA 3634
arctic_training/scheduler/__init__.py sha256=Br6Xr7bzfcnzYApSfZiqXq-asMCLntVzEJEyz55uBbM 694
arctic_training/scheduler/factory.py sha256=dNkvqIr3Q_J1nsCHV8BBmdXqIj_bliAsKahPbepe-XM 2445
arctic_training/scheduler/hf_factory.py sha256=SeEGkpUCK5asthl1AY1mnXbgW3hxpCXe4weNAVjeiYs 1315
arctic_training/synth/README.md sha256=u8lTftTfkt_O1c44XXW8FCX-tiO0oIq1xI3Cj_xC7XU 5270
arctic_training/synth/__init__.py sha256=LiI9Wkf8RI0z37Zhz9IK582Jhp5VZKQ9TS19WDUPArY 171
arctic_training/synth/base_caller.py sha256=uLTdriHPgkMURXXow18_Uvgghv3HBJxb5zLGX4Dy4W8 629
arctic_training/synth/callers.py sha256=82pzThZY-to6tZjPLeXaeCxHQROVyyJzChFNzLW0upc 5715
arctic_training/synth/cli.py sha256=hjKTl0HjkkPDhwYRAIiB8sDMVXYOv1LBrMLCN41xoAg 4413
arctic_training/synth/openai_callers.py sha256=q8Zk_5tfqgg1SCDiIb18Tlk4ZDJUfuimuFslbuMbCA4 16710
arctic_training/synth/test_cortex_caller.py sha256=0RZRYK50RKmn1Z6sHSTK7PvANPV1H-2gMsS_c-qpe9w 771
arctic_training/synth/test_openai_callers.py sha256=D8-9_L56G0XWrdupX5uurAbOgSXmy3VR73J54uT4tw0 1285
arctic_training/synth/test_vllm_caller.py sha256=rIpRkEUxDlrQejKvgJQRk_jv2CnxV51Wz8dSVmobB1M 496
arctic_training/synth/utils.py sha256=jOYtkV34JE_zSgqedsgTH1ssDCfdTN8HcaAY16mevao 144
arctic_training/tokenizer/__init__.py sha256=nLsgU9QSKPo5MM3zuFu6YOgyCTgAY2cylUZjcu5BFy8 694
arctic_training/tokenizer/factory.py sha256=3F73j0hs4yTTBuHIoBHezWm7V48pZLZkJ_ukkavYWSs 2456
arctic_training/tokenizer/hf_factory.py sha256=1ldxnwm6QlCEctWQW3XZorhDFT9ZixYB2lEBBR7O2EM 1331
arctic_training/trainer/__init__.py sha256=_XXwO0T8sJpJPWb9I_8OO229oAbKToBEedDNrXOtkvw 642
arctic_training/trainer/sft_trainer.py sha256=D0KhCaPZIataFUELeZwoJB7SczJ1qTEJ5KBHrjSgDYw 2132
arctic_training/trainer/trainer.py sha256=9ElYIP77NUvo9Z67dG7IUYEzwFRTJjbdThsPiaCx4WU 10056
arctic_training-0.0.1.dist-info/METADATA sha256=XU1d42cOJruX3NiKdMNkIHJM1m739gNiRr0OWkEkIgg 5012
arctic_training-0.0.1.dist-info/WHEEL sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc 105
arctic_training-0.0.1.dist-info/entry_points.txt sha256=ZZ17D5W3KDvamuNMFS1dY4e_L46ay46Z9XICE_fn4So 156
arctic_training-0.0.1.dist-info/licenses/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
arctic_training-0.0.1.dist-info/RECORD — —

entry_points.txt

[console_scripts]
arctic_synth = arctic_training.synth:main
arctic_training = arctic_training.cli:main
arctic_training_run = arctic_training.cli:run_script