Wheelodex » Projects » nemo-aligner

nemo-aligner

View on PyPI — Reverse Dependencies (0)

0.5.0 nemo_aligner-0.5.0-py3-none-any.whl

Wheel Details

Project: nemo-aligner
Version: 0.5.0
Filename: nemo_aligner-0.5.0-py3-none-any.whl
Download: [link]
Size: 145158
MD5: c8adf51e5640793bc1a8f750d29ecbc2
SHA256: 56f514f963e40dd0ab69c0bf535aa2cce1d040680a44e575af68b0b5f2bfd4d2
Uploaded: 2024-11-14 23:55:56 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: nemo_aligner
Version: 0.5.0
Summary: NeMo-Aligner - a toolkit for model alignment
Author: NVIDIA
Author-Email: nemo-toolkit[at]nvidia.com
Maintainer: NVIDIA
Maintainer-Email: nemo-toolkit[at]nvidia.com
Home-Page: https://github.com/NVIDIA/NeMo-Aligner
Download-Url: https://github.com/NVIDIA/NeMo-Aligner/releases
License: Apache2
Keywords: deep learning,machine learning,gpu,NLP,NeMo,nvidia,pytorch,torch,language,reinforcement learning,RLHF,preference modeling,SteerLM,DPO
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Intended Audience :: Information Technology
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Topic :: Scientific/Engineering :: Image Recognition
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development :: Libraries
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Classifier: Topic :: Utilities
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Environment :: Console
Classifier: Natural Language :: English
Classifier: Operating System :: OS Independent
Requires-Dist: jsonlines
Requires-Dist: megatron-core (>=0.8)
Requires-Dist: nemo-toolkit[nlp]
Requires-Dist: nvidia-pytriton
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 6024 characters]

WHEEL

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

RECORD

Path Digest Size
nemo_aligner/__init__.py sha256=aKT2OS3W_EfwiTyOfs8asVgiiR_MkBFR65PKc_a81tw 1146
nemo_aligner/package_info.py sha256=Q9MsQxcUU0x_FdJuQBlgZ8H7vcScOb1v9hsQXfcCFi8 1473
nemo_aligner/algorithms/__init__.py sha256=fAL6CiF1_taAoRVsurzSXhlfywLzVsXZOkEEtyBOfBQ 610
nemo_aligner/algorithms/critic_server_trainer.py sha256=j2NJ3ACfwEuMVFdy0LkJKFHpA846sehbZSnMxfj6lFo 14662
nemo_aligner/algorithms/dpo.py sha256=KUtEgTBFD1we4XjGcL3kasZLxHHCPYswSTwuUW6uNbQ 13306
nemo_aligner/algorithms/kto.py sha256=ypswPxZ8ueVqsuilQCviC371J0Us1o1nUGASDdfoQ8k 5295
nemo_aligner/algorithms/ppo.py sha256=DZbssFTeMDm-Tca8TlTtAds0k-YBNyXKV1rqKM41fu4 27490
nemo_aligner/algorithms/reward_server.py sha256=JaYk1XZ6FXhBgxV5vungvFieMb2N2SpNVvUVaF7zcXQ 4873
nemo_aligner/algorithms/rs.py sha256=k-DROXD7xugTGVY7MZ2cN_1G2OGEhZwqewp2aeSuTnU 19122
nemo_aligner/algorithms/spin.py sha256=MaEa6r8z63fPTigsPDVa7rvzY-0tOx2_e0hZn2NtEyM 23999
nemo_aligner/algorithms/supervised.py sha256=PR7aBent495J216jE0-_mnmhruKCoFI3Vbm_pZMv0cw 11637
nemo_aligner/data/__init__.py sha256=fAL6CiF1_taAoRVsurzSXhlfywLzVsXZOkEEtyBOfBQ 610
nemo_aligner/data/nlp/__init__.py sha256=fAL6CiF1_taAoRVsurzSXhlfywLzVsXZOkEEtyBOfBQ 610
nemo_aligner/data/nlp/builders.py sha256=TIKiF8frMHcLJHZLZE_Og6z7c4Bi-ECVf_PzWwAOAxc 15281
nemo_aligner/data/nlp/datasets.py sha256=iWl3nDTZFYLbUJgJygElFcUFKfqUpxnMRnWyNytRN7M 27811
nemo_aligner/metrics/__init__.py sha256=u4x5Omfo7L-5s80mxT7JUu0R_IU4F8oWU-8qefNekok 758
nemo_aligner/metrics/common.py sha256=CB3FAQE_CDbGB880tsK42BX10PHqQptufBu1rInuN6c 2061
nemo_aligner/models/__init__.py sha256=fAL6CiF1_taAoRVsurzSXhlfywLzVsXZOkEEtyBOfBQ 610
nemo_aligner/models/alignable_interface.py sha256=vrmG2ZsPqN4xon2ewoZB55fBzVu_L8HVfzY0RfXhlZs 3815
nemo_aligner/models/nlp/__init__.py sha256=fAL6CiF1_taAoRVsurzSXhlfywLzVsXZOkEEtyBOfBQ 610
nemo_aligner/models/nlp/gpt/__init__.py sha256=fAL6CiF1_taAoRVsurzSXhlfywLzVsXZOkEEtyBOfBQ 610
nemo_aligner/models/nlp/gpt/gpt_reward_model.py sha256=QgSClwjSK2oLrX8g0YzBmRs8gK7RyC_BdQCfgVVJqq0 9549
nemo_aligner/models/nlp/gpt/gpt_sft_model.py sha256=moDBe-zkq5vFlj_Byrdyvk4QE4efsvUnbnvY_Desuxo 10092
nemo_aligner/models/nlp/gpt/gpt_steerlm_model.py sha256=XchyWLRNaFRQPLriRnyArgQsbbR2aQx4e06zFUvh9vg 10357
nemo_aligner/models/nlp/gpt/megatron_gpt_critic.py sha256=zyhfOVWA6K1iUstKqbYe43YN-7dJRgIR8dCl3AzEB64 10173
nemo_aligner/models/nlp/gpt/megatron_gpt_dpo_model.py sha256=v8Cxi4a7zNWgXrKsXvS-6SOE-4ubUGZrcp5F1rMcHkE 21573
nemo_aligner/models/nlp/gpt/megatron_gpt_kto_model.py sha256=55LJmWG_erKBFegK83AKmAchKyjiL6xc_UCBfTortrc 19029
nemo_aligner/models/nlp/gpt/megatron_gpt_ppo_actor.py sha256=9gabz7XsQM3AsS0zMdTgpQeYYdWQInbrWUVLgppVe5s 17738
nemo_aligner/models/nlp/gpt/megatron_gpt_regression_reward_model.py sha256=g08z_8iPslK66di5EEtHxyfh_0WPpjQTyvvC4Sv5xWE 7136
nemo_aligner/models/nlp/gpt/megatron_gpt_reward_model.py sha256=1zyac8z7OldW3KjANKVppZ4fWAKHl14XROSFMabMdUg 20640
nemo_aligner/models/nlp/gpt/megatron_gpt_rs_actor.py sha256=w3iQpKwgknKptZw8_TkdFhUeX9gSTZS9qGPI6eQ1-Yw 10702
nemo_aligner/models/nlp/gpt/megatron_gpt_spin_model.py sha256=sZOFP6URbg73svNfj2SJp2J5KxayPzPuk3acf5J36o8 29511
nemo_aligner/models/nlp/gpt/reward_critic_clients.py sha256=OAdxQpHk5-LNRfHyVpZ1cDDAKet_s0MDYW2HZ-bng3w 7805
nemo_aligner/models/nlp/gpt/reward_model_classes.py sha256=PqHhRRKQKDgt_cswnXZOrmmAVSx5vvmkMKYrjoa7MJA 1143
nemo_aligner/servers/__init__.py sha256=fAL6CiF1_taAoRVsurzSXhlfywLzVsXZOkEEtyBOfBQ 610
nemo_aligner/servers/constants.py sha256=U0XqYVQp18Cii8-e6VY3E2OLGNxzn6TcCN4tjbXlt7Q 831
nemo_aligner/servers/http_communicator.py sha256=ddjXhRqqpeRTKCNcmGKAI2qA6w9ShuxdacrFA8GGJGM 2668
nemo_aligner/utils/__init__.py sha256=fAL6CiF1_taAoRVsurzSXhlfywLzVsXZOkEEtyBOfBQ 610
nemo_aligner/utils/batch_iterators.py sha256=hjVGHSgWIeLiWwPZL_IV6NlXTCc3lFcrHmTNIJpCpx4 4664
nemo_aligner/utils/distributed.py sha256=uXHjJSyWEvsNQ7eKQx49G-R24iEBle30OLYDCqkxWVo 17776
nemo_aligner/utils/parallel_state.py sha256=mFwKHfQTD8RZ2OR9FaFxZmjMJUtEGDMRi9US45GVG0k 3022
nemo_aligner/utils/ppo_utils.py sha256=CP_Nmlh-iGyJ8I4Po3-rFoCNmHtJUAqFE3cEtY46HXk 4432
nemo_aligner/utils/server_utils.py sha256=FulEbHJ1cPf1QTZo-_AW10PeE86RmWLOQWWr5_9t4qc 5104
nemo_aligner/utils/text_generation_utils.py sha256=a0CFnLXec6UJXPSqBFDRcv0dHbd1RHyCu6psGEC4jIM 5162
nemo_aligner/utils/train_script_utils.py sha256=QPinCRpd30YdMeuN0UAmaFVNs07QE9UIXQJQcoqc0zw 8529
nemo_aligner/utils/train_utils.py sha256=fpTU_YkQVgvQbWO5qZTHB9OtSERmDTjw_9jeOkxki6U 6421
nemo_aligner/utils/trainer_utils.py sha256=jSgmPfymTrNypgkjnU2zvrFSQQE9xEnQxzDjXGQH4UA 3256
nemo_aligner/utils/trt_llm.py sha256=-sjJruDldtk5UiGxRAajokR4AC1A0o6NPnMZSfBGoz4 11304
nemo_aligner/utils/utils.py sha256=D4tydkfawg6oUFf4PL1CKOtSTGBUjlIlA9xuD5jPCLM 18988
nemo_aligner-0.5.0.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
nemo_aligner-0.5.0.dist-info/METADATA sha256=nJsA7fOwyXEawGcARyoW5YqFuflVaSJ62gyStwgGnSA 7634
nemo_aligner-0.5.0.dist-info/WHEEL sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY 91
nemo_aligner-0.5.0.dist-info/top_level.txt sha256=HEZXOJAgbG3pZNZ0xCbTN0ED-jQsgayvNejqVjjo5Ik 13
nemo_aligner-0.5.0.dist-info/RECORD — —

top_level.txt

nemo_aligner