sapiens-accelerator

View on PyPIReverse Dependencies (0)

1.0.5 sapiens_accelerator-1.0.5-py3-none-any.whl

Wheel Details

Project: sapiens-accelerator
Version: 1.0.5
Filename: sapiens_accelerator-1.0.5-py3-none-any.whl
Download: [link]
Size: 207303
MD5: 41317897ef878689b0b5172f4ef2d409
SHA256: 06c214c4d7bca21acb32a617e911b9996b301d880d399f8acd20b4c9dbc01575
Uploaded: 2025-02-04 21:36:29 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: sapiens_accelerator
Version: 1.0.5
Author: OPENSAPI
Home-Page: https://github.com/
License: Proprietary Software
Requires-Python: >=3.9.0
Requires-Dist: numpy (==1.25.2)
Requires-Dist: packaging (==24.2)
Requires-Dist: psutil (==6.1.1)
Requires-Dist: PyYAML (==6.0.2)
Requires-Dist: torch (==2.4.1)
Requires-Dist: huggingface-hub (==0.28.1)
Requires-Dist: safetensors (==0.5.2)
Requires-Dist: deepspeed; extra == "deepspeed"
Requires-Dist: black (~=23.1); extra == "dev"
Requires-Dist: hf-doc-builder (==0.5.0); extra == "dev"
Requires-Dist: ruff (~=0.6.4); extra == "dev"
Requires-Dist: pytest (<=8.0.0,>=7.2.0); extra == "dev"
Requires-Dist: pytest-xdist (==3.6.1); extra == "dev"
Requires-Dist: pytest-subtests (==0.14.1); extra == "dev"
Requires-Dist: parameterized (==0.9.0); extra == "dev"
Requires-Dist: datasets (==3.0.1); extra == "dev"
Requires-Dist: diffusers (==0.32.2); extra == "dev"
Requires-Dist: evaluate (==0.4.3); extra == "dev"
Requires-Dist: torchdata (==0.10.1); extra == "dev"
Requires-Dist: torchpippy (==0.2.0); extra == "dev"
Requires-Dist: transformers (==4.45.2); extra == "dev"
Requires-Dist: scipy (==1.15.1); extra == "dev"
Requires-Dist: scikit-learn (==1.5.0); extra == "dev"
Requires-Dist: tqdm (==4.66.4); extra == "dev"
Requires-Dist: sapiens-machine (==1.0.9); extra == "dev"
Requires-Dist: timm (==1.0.14); extra == "dev"
Requires-Dist: rich; extra == "dev"
Requires-Dist: black (~=23.1); extra == "quality"
Requires-Dist: hf-doc-builder (==0.5.0); extra == "quality"
Requires-Dist: ruff (~=0.6.4); extra == "quality"
Requires-Dist: rich; extra == "rich"
Requires-Dist: sagemaker (==2.239.0); extra == "sagemaker"
Requires-Dist: datasets (==3.0.1); extra == "test-dev"
Requires-Dist: diffusers (==0.32.2); extra == "test-dev"
Requires-Dist: evaluate (==0.4.3); extra == "test-dev"
Requires-Dist: torchdata (==0.10.1); extra == "test-dev"
Requires-Dist: torchpippy (==0.2.0); extra == "test-dev"
Requires-Dist: transformers (==4.45.2); extra == "test-dev"
Requires-Dist: scipy (==1.15.1); extra == "test-dev"
Requires-Dist: scikit-learn (==1.5.0); extra == "test-dev"
Requires-Dist: tqdm (==4.66.4); extra == "test-dev"
Requires-Dist: sapiens-machine (==1.0.9); extra == "test-dev"
Requires-Dist: timm (==1.0.14); extra == "test-dev"
Requires-Dist: pytest (<=8.0.0,>=7.2.0); extra == "test-prod"
Requires-Dist: pytest-xdist (==3.6.1); extra == "test-prod"
Requires-Dist: pytest-subtests (==0.14.1); extra == "test-prod"
Requires-Dist: parameterized (==0.9.0); extra == "test-prod"
Requires-Dist: wandb (==0.19.5); extra == "test-trackers"
Requires-Dist: comet-ml (==3.48.1); extra == "test-trackers"
Requires-Dist: tensorboard (==2.18.0); extra == "test-trackers"
Requires-Dist: dvclive (==3.48.1); extra == "test-trackers"
Requires-Dist: pytest (<=8.0.0,>=7.2.0); extra == "testing"
Requires-Dist: pytest-xdist (==3.6.1); extra == "testing"
Requires-Dist: pytest-subtests (==0.14.1); extra == "testing"
Requires-Dist: parameterized (==0.9.0); extra == "testing"
Requires-Dist: datasets (==3.0.1); extra == "testing"
Requires-Dist: diffusers (==0.32.2); extra == "testing"
Requires-Dist: evaluate (==0.4.3); extra == "testing"
Requires-Dist: torchdata (==0.10.1); extra == "testing"
Requires-Dist: torchpippy (==0.2.0); extra == "testing"
Requires-Dist: transformers (==4.45.2); extra == "testing"
Requires-Dist: scipy (==1.15.1); extra == "testing"
Requires-Dist: scikit-learn (==1.5.0); extra == "testing"
Requires-Dist: tqdm (==4.66.4); extra == "testing"
Requires-Dist: sapiens-machine (==1.0.9); extra == "testing"
Requires-Dist: timm (==1.0.14); extra == "testing"
Provides-Extra: deepspeed
Provides-Extra: dev
Provides-Extra: docs
Provides-Extra: quality
Provides-Extra: rich
Provides-Extra: sagemaker
Provides-Extra: test_dev
Provides-Extra: test_prod
Provides-Extra: test_trackers
Provides-Extra: testing
License-File: LICENSE.txt
[No description]

WHEEL

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

RECORD

Path Digest Size
sapiens_accelerator/__init__.py sha256=va7AYCB_KF0rxz_kQUi6nzSoHvwUEmMFkpfhCz-FN4I 2047
sapiens_accelerator/accelerator.py sha256=ssVDj6ydG7TJFa9cj_o35qFBz6qq5jaFKJCOz_DR83s 80386
sapiens_accelerator/big_modeling.py sha256=soly4n2bMgWWPdfJLQ4ebjOVV4CVDnbz39CeCsdxzos 13132
sapiens_accelerator/checkpointing.py sha256=uqouGR3wsyK3yPbrWPkJHZhckLNzlh50PT9FTc9US8M 8926
sapiens_accelerator/data_loader.py sha256=uoAhkstBUQhXg-0dMHgy5WbpBINF0s1_enXOabyQOJw 31931
sapiens_accelerator/hooks.py sha256=ryp0vudHUdSKxtuPC5owykeHcKLV9LS2yN3sk8NUBto 16400
sapiens_accelerator/inference.py sha256=98p6SubkpBL1s5bdtx4bBdoAQpumVX4K4s_i3hPuqzA 5545
sapiens_accelerator/launchers.py sha256=nzSOMt0yDd-nng_zqEbuEw6mN8zdArGYQV-LJBh2s00 7931
sapiens_accelerator/local_sgd.py sha256=Yy9nCDLN9xiqimtXKVDRnicYziVCe524BPMz8y6Y_aM 2819
sapiens_accelerator/logging.py sha256=IRmkF1fYGyZvjWKrBi_K-f33l198VLL9c3Awt8ZN53M 2986
sapiens_accelerator/memory_utils.py sha256=y6WO1S-KYBBVRdIpvTWJzA_ZeeG6N3riO03NM6mR-JA 625
sapiens_accelerator/optimizer.py sha256=khnuuz3ULG6oXMRwHODssc3lKg0tHhAwRj9dgxNEEdY 6886
sapiens_accelerator/scheduler.py sha256=nO90sySSLvJXmyKRTrv3xQ1XlE4wzldzkC8etye7ER4 2917
sapiens_accelerator/state.py sha256=WI_qhaJJZgOO3CysQkGPhkvSp_DjUcwvobvI3l3IVJg 28362
sapiens_accelerator/tracking.py sha256=jKb2CN3KAQYUaubGA-FMPXZstORaau6pPDhKSMpZ33M 14920
sapiens_accelerator/commands/__init__.py sha256=y6WO1S-KYBBVRdIpvTWJzA_ZeeG6N3riO03NM6mR-JA 625
sapiens_accelerator/commands/accelerate_cli.py sha256=t0xJeOzicJdM-SIU5WgHgmXqSmiRoHCGMO9nvYX1C0Q 2537
sapiens_accelerator/commands/env.py sha256=ysjEy2fdZMZkWifyECAPTG_PSBOhJoU8rHscvrqMrAI 4056
sapiens_accelerator/commands/estimate.py sha256=6-3za6MO5OTzgS0YQCnpAq-V0v-2iHaME6maTVKVAdE 9369
sapiens_accelerator/commands/launch.py sha256=wLEPcM4NfPUaTxcOwcPmJXvRzPQZjpQdL2uuR-GVMvI 19977
sapiens_accelerator/commands/merge.py sha256=b8v8VwmXKbuAohxsWXSRH8pxb3NomyLE9qOYgUd_Zrk 2878
sapiens_accelerator/commands/test.py sha256=1CJTc1lnpIYpKKiLTXDDiYuVfpXczNbP4v8Ejy-uONY 1903
sapiens_accelerator/commands/tpu.py sha256=O6HV1OZBeU2tT8lnLSkrO7plR3U67F3xzkLbpKihZzI 5310
sapiens_accelerator/commands/utils.py sha256=aeu5IWsUVCN-ElWM-7VlUeX_5iiN_TksuKOMPTfg88Y 3533
sapiens_accelerator/commands/config/__init__.py sha256=VzqZJ3NhvN4cCc2gA7rQyoLgJzWHytSRZzve2M9EKZM 2130
sapiens_accelerator/commands/config/cluster.py sha256=PeqccHn2MpwN3fsMVQzo6C3AiB7NlChKQ0h4JTX1dCI 27198
sapiens_accelerator/commands/config/config.py sha256=KuW5T5BzgEDdzqmtmjFR_QEMgQAVwqGCtsHqO2CUTkU 3389
sapiens_accelerator/commands/config/config_args.py sha256=aHJRhvcCPGyFvNaIvV4ouABu5uhUM4XuJHuLCGrmKFs 9267
sapiens_accelerator/commands/config/config_utils.py sha256=udWmSJeTgK9XKwD4ZzFxPteu-KVbUjkcXbbDebAvrXs 3457
sapiens_accelerator/commands/config/default.py sha256=JR7-30dhrLjwNl-lR-ZEp9NL8IPwoQCx3TyYl_ZB9KY 4817
sapiens_accelerator/commands/config/sagemaker.py sha256=9EYXRAfnA6a4HA5DURpXqrnzpNwqp42usfBJrLC-Vfg 8482
sapiens_accelerator/commands/config/update.py sha256=4Bt9lej5rjN0yeoiSE6v1CFdqkJgPOW7PxNmhpR1pOE 2790
sapiens_accelerator/commands/menu/__init__.py sha256=rpofe71Tgg_Lli494Rmo-Sa5JR3Y9z7OmrWlb73WLhs 1289
sapiens_accelerator/commands/menu/cursor.py sha256=w5IDQ0drKBJvsbsDDFi0wVaTGkmb4FPufpwq4CGYpOY 2400
sapiens_accelerator/commands/menu/helpers.py sha256=ZeieJ1taPfRqa6aBLGcRrx3o4gerjyGmwTbMfseQses 1970
sapiens_accelerator/commands/menu/input.py sha256=PJk_8yLsw_UCpJHGba5XJVT0pkt7MCKkw7a44uhDJhY 2566
sapiens_accelerator/commands/menu/keymap.py sha256=FOAWuPQvHKoO0ar1agR5VtEYRAdpzOZacU6-vZF0pWk 4002
sapiens_accelerator/commands/menu/selection_menu.py sha256=QnkxA61eLwSXNaE4oRs-VMIXTuvVReixJahINtNbJLQ 4947
sapiens_accelerator/test_utils/__init__.py sha256=VDg2pVc5uEKwDEL7IcyR-BRQ1nZsWfNevELLGuqrhcE 2056
sapiens_accelerator/test_utils/examples.py sha256=hd-abMdo3g4nX_141J-Tuj4Rham0tzJWY0_NB1dTcg8 4810
sapiens_accelerator/test_utils/testing.py sha256=lkYDgKhAvhBiJ4TsvCvwXZRi-_kgDI3fxiF_QXoPhMA 13936
sapiens_accelerator/test_utils/training.py sha256=5wb8xLZiPAocuOAyN5ElDQCbzfO5D8HB6I1N-YyTowQ 4078
sapiens_accelerator/test_utils/scripts/__init__.py sha256=y6WO1S-KYBBVRdIpvTWJzA_ZeeG6N3riO03NM6mR-JA 625
sapiens_accelerator/test_utils/scripts/test_cli.py sha256=IxA217gxXGqEQPqAt_TlzZS5L4VZw_2WFUcROvQJJtc 1427
sapiens_accelerator/test_utils/scripts/test_ddp_comm_hook.py sha256=w7c5KzLCWz-aocbrlyzEWNMT0M3ike5xvQl-vzfM6fM 3590
sapiens_accelerator/test_utils/scripts/test_distributed_data_loop.py sha256=5p9_x-qmuUXOPoE_T7wmCfRnYs4ZvjoMYup7hiHGnF0 12856
sapiens_accelerator/test_utils/scripts/test_merge_weights.py sha256=TmGsQbXOM3NLvXSiKMF8QTGE0zrcq26796HxgMwQKTg 6216
sapiens_accelerator/test_utils/scripts/test_notebook.py sha256=qXQIsnB9xsaFv2Ai5xFgmZcbyUXUm3UsnxiZzclojGg 4002
sapiens_accelerator/test_utils/scripts/test_ops.py sha256=y79ww4_zT9_CfM6ui7P_VULsuwAGUJgoRc4yzpcvSZw 5649
sapiens_accelerator/test_utils/scripts/test_script.py sha256=YX4xW0saj1395yugfxDWYAI7TRcv2gONdqn816isGgg 27827
sapiens_accelerator/test_utils/scripts/test_sync.py sha256=20hiFIV-f26_Dxx5eJlRtLHDuZUqby1lfaavybU5aPE 13844
sapiens_accelerator/test_utils/scripts/external_deps/__init__.py sha256=y6WO1S-KYBBVRdIpvTWJzA_ZeeG6N3riO03NM6mR-JA 625
sapiens_accelerator/test_utils/scripts/external_deps/test_checkpointing.py sha256=XknIm3h-rqTH5BiNzzgn2gm-3qMs2Y3C1cKTkJxs9bA 9018
sapiens_accelerator/test_utils/scripts/external_deps/test_ds_multiple_model.py sha256=ceLR0ABU1fkhQTbPCCWrb10ItEaXY6JbVHMdhE_quGY 11531
sapiens_accelerator/test_utils/scripts/external_deps/test_metrics.py sha256=h04EK9Ez3kALudTOI_MtnGcK2tdgrdRjmznA0fdo2So 10667
sapiens_accelerator/test_utils/scripts/external_deps/test_peak_memory_usage.py sha256=RoFOBmgCN0rF0_ba8J6SA8groGYeU1rMXvyrVGbSii4 9417
sapiens_accelerator/test_utils/scripts/external_deps/test_performance.py sha256=WE_Z59ms60uxgkz-9MtSg87RYWx_TDAHv0JT6ko7iyo 8057
sapiens_accelerator/test_utils/scripts/external_deps/test_pippy.py sha256=0XZWyIUs7v9HacjReWWJqUODEbNbCE3KBXusysrwEWE 3623
sapiens_accelerator/test_utils/scripts/external_deps/test_zero3_integration.py sha256=TOC2XX1e9bEhArAHrSFjq47nhrMzV-aAguQaIBMTn8s 2160
sapiens_accelerator/utils/__init__.py sha256=cnGQEbUQpJXulzsJOu7eRXjaszYXfLd17dAqkU9c7b0 6809
sapiens_accelerator/utils/constants.py sha256=cNfYv3CEj3oZGZIWtEmX-Bbu8X5aXinDaiLOg2bSljc 3380
sapiens_accelerator/utils/dataclasses.py sha256=ANYxzfL94eGnF8wM3IqxfWaPybghEp4294ZJgejABkI 72936
sapiens_accelerator/utils/deepspeed.py sha256=sNh0XXxZ_IHP1p1WmqGvvN_h1MShDhOxPwuMAU8kuMo 6061
sapiens_accelerator/utils/environment.py sha256=qNu9NJEstXKXOlZ-6xtDoRn5i2z9YydH0acc1YGClbI 6597
sapiens_accelerator/utils/fsdp_utils.py sha256=Kaom5rNeBevbjnI-7TLLAAFJixiePopjWocYGZjC4W0 12404
sapiens_accelerator/utils/imports.py sha256=fzlkBlbfxlhKe2J6FCATHUDRb3FqGUc7oYtqw5MSU80 11228
sapiens_accelerator/utils/launch.py sha256=iVoccHtv6ckP3cs6cosuOuShzZhMa-cygO_s_lty_Pw 25261
sapiens_accelerator/utils/megatron_lm.py sha256=HQnx7boQc4pyq2hlox6b5cLMuqr55Cr0xFXwJi7RIgU 45086
sapiens_accelerator/utils/memory.py sha256=BhxQWqIUlm3jZHhJTLiJzEs8C3TR216KwWsMEhnXhvA 4100
sapiens_accelerator/utils/modeling.py sha256=MP3kBlmf3wYgI2Zrbp17G9qpt68Q_Oa5ext8yFKjhkY 48777
sapiens_accelerator/utils/offload.py sha256=5pfD96yrb6iAT3ELsToRNzZbPky3rzI1rPmlQsPweCs 5932
sapiens_accelerator/utils/operations.py sha256=IcKZyLEpFZJPQ-WJYlEReiS2luU0ZESk0jLG9LnwXYk 17938
sapiens_accelerator/utils/other.py sha256=o9Pna4Yy8WRZ4WLTh9n0ZjV01Ugn97i2QWC8HjXzNz4 7214
sapiens_accelerator/utils/random.py sha256=uI3T_FtOSkpYoZixE1VGg35k6SQPhdBvyVxnqhN3OK4 5143
sapiens_accelerator/utils/rich.py sha256=4O7Nf3ClMsTXPPfmzoq-xvCDvWkZqqTZtjth-MP--u4 1506
sapiens_accelerator/utils/sapiens.py sha256=M-kHI9uEcX5eUj11KyJFMy07QsoI1MgzdAo8faDxH7o 12880
sapiens_accelerator/utils/torch_xla.py sha256=Ri4_mZeZSmtF1dLrS3vuR77YurSwp6Rc8F01-heAgi8 2117
sapiens_accelerator/utils/tqdm.py sha256=9289UU7rQKHPVRVES6pq4Dfr48KjClPI7-cO58dYfPQ 2039
sapiens_accelerator/utils/transformer_engine.py sha256=PsevogD73wFF_RS0FKlEARUCQ3aJ-hjk9k_ULW84Wnw 5390
sapiens_accelerator/utils/versions.py sha256=AVFwbfY0Y6xYk8URxYc__C8RbmusOHV-eKoPkOdnJEM 2088
sapiens_accelerator-1.0.5.dist-info/LICENSE.txt sha256=WqB2vIA5tH5lqLTr53yT_oy1m0wYfuvCPQKxdDHWimg 115
sapiens_accelerator-1.0.5.dist-info/METADATA sha256=S-a4thlc5QcTsvK8na5a6UodWdh0pJnpCwJZi_MxaPc 3717
sapiens_accelerator-1.0.5.dist-info/WHEEL sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ 91
sapiens_accelerator-1.0.5.dist-info/entry_points.txt sha256=tp8Jxhcsn2vYEXpyAotYyxI__nBdXSBBYpsmQ0UfmMk 341
sapiens_accelerator-1.0.5.dist-info/top_level.txt sha256=Pz_ZfPdaP_kCBk9SZGvMee8ggPnh0NSyI0r5BiBojrc 20
sapiens_accelerator-1.0.5.dist-info/RECORD

top_level.txt

sapiens_accelerator

entry_points.txt

sapiens-accelerator = accelerate.commands.accelerate_cli:main
sapiens-accelerator-config = accelerate.commands.config:main
sapiens-accelerator-estimate-memory = accelerate.commands.estimate:main
sapiens-accelerator-launch = accelerate.commands.launch:main
sapiens-accelerator-merge-weights = accelerate.commands.merge:main