Wheelodex » Projects » flex-model

flex-model

View on PyPI — Reverse Dependencies (0)

0.1.0 flex_model-0.1.0-py3-none-any.whl

Wheel Details

Project: flex-model
Version: 0.1.0
Filename: flex_model-0.1.0-py3-none-any.whl
Download: [link]
Size: 62107
MD5: 9ba7c75bc1d9c6cf0cdf3295201e41b9
SHA256: d8e5a1c3c3df69692faa4e24c87d050a642d58374d7c3f07fca6fa567de3a1b7
Uploaded: 2023-12-05 15:47:58 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: flex-model
Version: 0.1.0
Summary: FlexModel - A Framework for Interpretability of Distributed Large Language Models
Author: Matthew Choi
Author-Email: matthew.choi[at]vectorinstitute.ai
Maintainer: Matthew Choi
Maintainer-Email: matthew.choi[at]vectorinstitute.ai
Home-Page: https://github.com/VectorInstitute/flex_model
License: mit
Requires-Dist: torch (>=2.1)
Requires-Dist: torch (>=2.1); extra == "all"
Requires-Dist: fairscale; extra == "all"
Requires-Dist: pytest; extra == "all"
Requires-Dist: sentencepiece; extra == "all"
Requires-Dist: submitit; extra == "all"
Requires-Dist: tokenizers; extra == "all"
Requires-Dist: transformers; extra == "all"
Requires-Dist: coverage; extra == "all"
Requires-Dist: pre-commit; extra == "all"
Requires-Dist: ruff; extra == "all"
Requires-Dist: sphinx-rtd-theme; extra == "all"
Requires-Dist: matplotlib; extra == "all"
Requires-Dist: torch (>=2.1); extra == "core"
Requires-Dist: torch (>=2.1); extra == "dev"
Requires-Dist: fairscale; extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: sentencepiece; extra == "dev"
Requires-Dist: submitit; extra == "dev"
Requires-Dist: tokenizers; extra == "dev"
Requires-Dist: transformers; extra == "dev"
Requires-Dist: coverage; extra == "dev"
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: ruff; extra == "dev"
Requires-Dist: sphinx-rtd-theme; extra == "dev"
Requires-Dist: torch (>=2.1); extra == "examples"
Requires-Dist: fairscale; extra == "examples"
Requires-Dist: sentencepiece; extra == "examples"
Requires-Dist: tokenizers; extra == "examples"
Requires-Dist: transformers; extra == "examples"
Requires-Dist: matplotlib; extra == "examples"
Requires-Dist: torch (>=2.1); extra == "test"
Requires-Dist: fairscale; extra == "test"
Requires-Dist: pytest; extra == "test"
Requires-Dist: sentencepiece; extra == "test"
Requires-Dist: submitit; extra == "test"
Requires-Dist: tokenizers; extra == "test"
Requires-Dist: transformers; extra == "test"
Provides-Extra: all
Provides-Extra: core
Provides-Extra: dev
Provides-Extra: examples
Provides-Extra: test
License-File: LICENSE
[No description]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.42.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
_test/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
_test/conftest.py sha256=8qmc14GlM5PDBbNRj35YuC1XbrmQar0tBhPB5saS84s 1624
_test/core/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
_test/core/test_hook_function.py sha256=joQ_tZBHvIjcbvOnBPNAsUsKSF9jp21YA-WtqspEhmY 6624
_test/core/test_hook_function_group_manager.py sha256=CNLS6pBn6ypwPT6s6yoFDdMSxg3QXDxT40cgRJMPyXU 6453
_test/core/test_wrapper.py sha256=Jh7DIQqJSOeLKBfbAwmLYCi3DIUBbXBLQN58UdjUum4 8719
_test/distributed/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
_test/distributed/test_initialize.py sha256=trRyH0JaqJedivOtuyeKKM3og0-7Arw-V0yONba5r_A 1111
_test/multi_gpu/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
_test/multi_gpu/registry.py sha256=LoDUyrf3E8MZO5H2zFIUc2lnSVpH0n3Dfo09q9LlwWM 1887
_test/multi_gpu/run_multi_gpu_tests_slurm.py sha256=qDAAPK_XvXWmW_h89ixqj4TMrT4u60dNaDK5-VacsdQ 4406
_test/multi_gpu/testing_utils.py sha256=ZTFZKCL_k3HVEAmkAz1Uv5B8Ycm1rZJ-X7Yv_Ramgx8 10197
_test/multi_gpu/core/__init__.py sha256=1tuvz6GNc9SVb9RlyqPoIDD-AdGsfsfsFa-ZpmumWkc 415
_test/multi_gpu/core/test_dp_wrappers_vs_single_gpu.py sha256=BEG94hkESv6BHR_ZJojiosmAywv0rNnrlhAyl6Dcpeg 7162
_test/multi_gpu/core/test_fairscale_mpu_vs_ddp.py sha256=akR4cq8QGV7bXr6zypqeXTNFPrjcj3yws_itu2m_E_Y 7948
_test/multi_gpu/distributed/__init__.py sha256=HyfeXEkAyFO0LZT6DloZdHutO5Mb7pvdmiaTpvBq2r4 431
_test/multi_gpu/distributed/test_initialize.py sha256=Eub0wEaWlcSV47DDV532L8_7v8PEG2U7rsngoxzgTRo 3049
_test/multi_gpu/distributed/test_mappings.py sha256=ovyl-yx1AWMW0tszhsQl4sC2vyWvbPueaELsdBx1vYs 6886
_test/traverse/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
_test/traverse/test_nodes.py sha256=aEFz01D5MaE19sKp5dKR-2CLhRtS3fLbqObZ--hKHA4 875
_test/traverse/test_ops.py sha256=IOf6tgtJl-ooqQqdP6_AsGAKhuD5_JJ0IKxw4aOc8mI 650
demos/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
demos/induction_heads_multigpu.py sha256=4mWQZB6cAHxxZ3PoPYZ7sl39kkdK5kPaWgGoyyxpmOs 9463
examples/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
examples/fsdp_example.py sha256=WGMUddSF-syT6lGF_gnRJbO8Rr8K8WwXcvUYC1Gqlno 5734
examples/megatron_example.py sha256=s48-2sLWWbg81EgoAZfpN5uvqwpO3XRD4fj5_g96jHw 3193
examples/single_gpu_example.py sha256=OYlCgDLoi4COFCyVjmrIfuPyExIu6J9BMmGaUWodnnY 2058
flex_model/__init__.py sha256=iVvI2bFyyQZC6V9vqbb7nd8Eqr0EMh8Lr3VC7iF4KZE 42
flex_model/package_info.py sha256=MGUCuSdXvK2UAq3vgGviY1soHY5s0kYFXJRTKvgcBDY 491
flex_model/utils.py sha256=5LcBIGoORulWTjyvv0feo5TshvTgeI51Y-0ptyo7x1E 172
flex_model/core/__init__.py sha256=UzflZ36DPvTrkUT4wPZ9N-_ZYBePsZpMg9DT8XWIw74 107
flex_model/core/core_utils.py sha256=M6oBXanBAnLbokpZ0qcWw6Z9E6hNv2MyEX2elRKgtp0 674
flex_model/core/hook_function.py sha256=GHVsfrQZNqJjawafBBd_fZn_QQUKLp6bSyuBWfTo-qQ 15565
flex_model/core/wrapper.py sha256=Wqky5MkITe2TjJcn-SQG4m3iUpqMgfoR0Ua0XBIu2Gk 27582
flex_model/distributed/__init__.py sha256=w_nHk4WMQ82bSjhcK8KpSEgqtRrY4NR3KhZKkO51e2A 727
flex_model/distributed/distributed_state.py sha256=IKfnyXRm_yvZf4KHa1C4Htf14A5HWQHN1VPDuQtFAqE 16481
flex_model/distributed/mappings.py sha256=OPrm4VqftSko3vya-_pc36Qj-GFPhUUxfNdPBCQ7G8s 16049
flex_model/distributed/strategies.py sha256=NQ3y5K7kksUSwTeZiRilrtskDDOr1AOOPD7nIH6nJRY 8746
flex_model/traverse/__init__.py sha256=plWRIeENOjp8hI8QKD-L0OUe1XqQqX7FonrPP5UPBXA 161
flex_model/traverse/nodes.py sha256=epjcKtKMYSsRwx-ZLtdiZcWxG_SA8xMM2rfw2Le5mdY 10152
flex_model/traverse/ops.py sha256=DHvHgOeAMWKhGO1aSAhl0Kf9UN9qUnDKwieOm6yYmoY 3511
profiling/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
profiling/profile_hooks.py sha256=_hah-abq4GyrRj20rjg6Z5vJ2iL7eJDlD5VpalVkF60 7645
profiling/utils.py sha256=--K8pDNjti4l8KzQ_TJxAFn0xNmyyYvOmjAFLaSnwn0 12278
flex_model-0.1.0.dist-info/LICENSE sha256=yaiq2te9ivpvgOd0AHBa9g17OICLwTlzZgOHc0RiFSI 1069
flex_model-0.1.0.dist-info/METADATA sha256=dWPwJT4_aCdHDGPXGIWuivQ5Pv9BWuE36uWELg5P8-s 2142
flex_model-0.1.0.dist-info/WHEEL sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM 92
flex_model-0.1.0.dist-info/top_level.txt sha256=-4djQJC59ApDtMaEIc-_Bp1kwG2bFCzeqR53rcFcQrE 42
flex_model-0.1.0.dist-info/RECORD — —

top_level.txt

_test
demos
examples
flex_model
profiling