经 AI Skill Hub 精选评估,极简LLM预训练 获评「推荐使用」。这款AI工具在功能完整性、社区活跃度和易用性方面表现出色,AI 评分 7.5 分,适合有一定技术背景的用户使用。
极简LLM预训练 是一款基于 Python 开发的开源工具,专注于 llm、optimizers、pretraining 等核心功能。作为 GitHub 开源项目,它拥有活跃的社区支持和持续的版本迭代,代码完全透明可审计,支持本地部署以保护数据隐私。无论是个人使用还是集成到企业工作流,都能提供稳定可靠的解决方案。
极简LLM预训练 是一款基于 Python 开发的开源工具,专注于 llm、optimizers、pretraining 等核心功能。作为 GitHub 开源项目,它拥有活跃的社区支持和持续的版本迭代,代码完全透明可审计,支持本地部署以保护数据隐私。无论是个人使用还是集成到企业工作流,都能提供稳定可靠的解决方案。
# 方式一:pip 安装(推荐)
pip install minimalist_llm_pretraining
# 方式二:虚拟环境安装(推荐生产环境)
python -m venv .venv
source .venv/bin/activate # Windows: .venv\Scripts\activate
pip install minimalist_llm_pretraining
# 方式三:从源码安装(获取最新功能)
git clone https://github.com/OptimAI-Lab/Minimalist_LLM_Pretraining
cd Minimalist_LLM_Pretraining
pip install -e .
# 验证安装
python -c "import minimalist_llm_pretraining; print('安装成功')"
# 命令行使用
minimalist_llm_pretraining --help
# 基本用法
minimalist_llm_pretraining input_file -o output_file
# Python 代码中调用
import minimalist_llm_pretraining
# 示例
result = minimalist_llm_pretraining.process("input")
print(result)
# minimalist_llm_pretraining 配置文件示例(config.yml) app: name: "minimalist_llm_pretraining" debug: false log_level: "INFO" # 运行时指定配置文件 minimalist_llm_pretraining --config config.yml # 或通过环境变量配置 export MINIMALIST_LLM_PRETRAINING_API_KEY="your-key" export MINIMALIST_LLM_PRETRAINING_OUTPUT_DIR="./output"
<a href="https://arxiv.org/abs/2506.16659"> <img src="https://img.shields.io/static/v1?label=arXiv&message=2506.16659&color=b31b1b" /> </a>
Preliminary code release for our paper "Memory-Efficient LLM Pretraining via Minimalist Optimizer Design", by Athanasios Glentis, Jiaxiang Li, Andi Han and Mingyi Hong.
The SCALE optimizer code can be found in mem_eff_pt/pt_scale/scale_optimizer.py. The repository provides the scripts used in our experiments in the scripts/ directory.
Example script (350m_scale.sh):
torchrun --standalone --nproc_per_node 4 torchrun_main_DDP.py \
--model_name 350m_scale \
--model_config configs/llama_350m.json \
--optimizer scale \
--lr 1e-3 \
--momentum 0.9 \
--weight_decay 0.0 \
--batch_size 128 \
--total_batch_size 512 \
--num_training_steps 60000 \
--warmup_steps 6000 \
--dtype bfloat16 \
--eval_every 1000 \
--save_every 99999 \
--seed 42 \
--scheduler cosine \
--dataset_path /path/to/c4/en \
高效的LLM预训练工具,值得关注
该工具未明确声明开源协议,商业使用前请联系原作者确认授权范围,避免侵权风险。
AI Skill Hub 为第三方内容聚合平台,本页面信息基于公开数据整理,不对工具功能和质量作任何法律背书。
建议在沙箱或测试环境中充分验证后,再部署至生产环境,并做好必要的安全评估。
AI Skill Hub 点评:极简LLM预训练 的核心功能完整,质量良好。对于AI 技术爱好者来说,这是一个值得纳入个人工具库的选择。建议先在非生产环境试用,再逐步推广。
| 原始名称 | Minimalist_LLM_Pretraining |
| 原始描述 | 开源AI工具:ICML 2026 Memory-Efficient LLM Pretraining via Minimalist Optimizer Design 。⭐15 · Python |
| Topics | llmoptimizerspretrainingpython |
| GitHub | https://github.com/OptimAI-Lab/Minimalist_LLM_Pretraining |
| 语言 | Python |
收录时间:2026-05-26 · 更新时间:2026-05-30 · License:未公布 · AI Skill Hub 不对第三方内容的准确性作法律背书。