用途:MolMIM NIM 用于小分子的潜在空间生成与优化,可对种子 SMILES 进行生成、采样、嵌入(embedding)、隐藏状态操作与解码,适合分子设计与性质优化场景。
主要能力:托管 API 暴露 /generate,支持种子 SMILES 生成或优化;algorithm 取 "CMA-ES" 时进行引导式性质优化,取 "none" 时进行无引导采样。property_name 支持 "QED" 或 "plogP"。本地容器额外暴露 /embedding、/hidden、/decode、/sampling,构成更完整的潜在空间工作流。
使用方式:托管请求使用 Authorization: Bearer $NGC_API_KEY,端点 https://health.api.nvidia.com/v1/biology/nvidia/molmim/generate。本地 Docker 使用镜像 nvcr.io/nim/nvidia/molmim:1.0.0,需将 LOCAL_NIM_CACHE 挂载到 /home/nvs/.cache/nim,启动前设置 NGC_CLI_API_KEY(或由 NGC_API_KEY/NVIDIA_API_KEY 映射)。就绪检查:until curl -sf http://localhost:8000/v1/health/ready; do sleep 5; done。本地推理无需 Authorization 头。
参数与限制:字段名为 smi 而非 smiles;num_molecules 为 1-100,iterations 为 1-1000,particles 为 2-1000,min_similarity 为 0-1,scaled_radius 为 0-2 且主要用于 algorithm: "none" 或本地 /sampling。托管 API 不提供 /embedding、/hidden、/decode、/sampling,调用会返回 404。401 表示 NGC 密钥缺失或未授权,422 表示 SMILES 无效、算法或 property_name 不支持、参数越界。
输出与验证:生成结果为候选分子,非已验证命中,需用 RDKit 等检查可解析性、唯一性、性质范围与相似度约束,并做下游性质、对接、亲和力、毒性与合成可行性验证。托管 /generate 可能返回 {sample, score} 对象的 JSON 字符串,本地端点可能返回 generated,需同时解析两种格式。
适用场景:小分子生成、性质引导优化、潜在空间嵌入与解码研究。具体收费、价格与平台信息暂未提供。