SAVRN
Search Contact SAVRN

Open-weight model · Robotics

SOMA-X

by NVIDIA nvidia/SOMA-X

SOMA (Unifying Parametric Human Body Models) is a unified framework that decouples identity representation from pose parameterization by mapping supported parametric models to canonical body and hand topologies and skeletons, enabling shared Linear Blend…

Parameters
Context
Weights796.3 MB
Licenseapache-2.0
AccessOpen weights
Monthly Downloads

Model Card

By NVIDIA, published under apache-2.0, revision 104578ed5885.

SOMA (Unifying Parametric Human Body Models) is a unified framework that decouples identity representation from pose parameterization by mapping supported parametric models to canonical body and hand topologies and skeletons, enabling shared Linear Blend Skinning (LBS) pipelines across backends. The full-body layer supports six identity backends (SOMA-shape, SMPL, SMPL-X, MHR, ANNY, and GarmentMeasurements). SOMA-X v0.3 also includes wrist-local left/right hand layers with native SOMA identity and articulation priors plus interoperability with user-supplied MANO models. This model is ready for commercial use. SOMA is released under the Global SOMA is intended for use by computer vision…

Read NVIDIA's full model card

Model Overview

Description:

SOMA (Unifying Parametric Human Body Models) is a unified framework that decouples identity representation from pose parameterization by mapping supported parametric models to canonical body and hand topologies and skeletons, enabling shared Linear Blend Skinning (LBS) pipelines across backends. The full-body layer supports six identity backends (SOMA-shape, SMPL, SMPL-X, MHR, ANNY, and GarmentMeasurements). SOMA-X v0.3 also includes wrist-local left/right hand layers with native SOMA identity and articulation priors plus interoperability with user-supplied MANO models.

This model is ready for commercial use.

License/Terms of Use:

SOMA is released under the Apache 2.0 License.

Deployment Geography:

Global

Use Case:

SOMA is intended for use by computer vision researchers, graphics and animation engineers, machine learning engineers, and robotics researchers and companies. Specific use cases include: - Pose estimation and human reconstruction — unified pose interface enables seamless identity substitution across backends without retraining. - Motion generation and animation — apply motion capture sequences to any supported identity model using the same axis-angle pose parameterization. - Avatar synthesis and digital humans — freely mix identity sources with SOMA's pose representation. - Simulation and robotics — lightweight analytical forward pass enables real-time simulation pipelines with diverse body shapes. - Hand animation and reconstruction — wrist-local left/right hand layers support mid, low, and extra-low LODs and a sampled articulation prior.

Release:

SOMA-X v0.3.0

Reference(s):

  • SOMA: Unifying Parametric Human Body Models.
  • SMPL: A Skinned Multi-Person Linear Model — Loper et al., 2015
  • SMPL-X: Expressive Body Capture: 3D Hands, Face, and Body from a Single Image — Pavlakos et al., 2019
  • MHR: Momentum Human Rig, Meta
  • ANNY: Anthropometric body model spanning full human lifespan

Model Architecture:

Architecture Type: Analytical / Parametric; optional shallow Multilayer Perceptron (MLP) for pose-dependent surface correctives

Network Architecture: The core pipeline uses no learned neural network components. It is composed of three closed-form analytical modules: 1. Barycentric Mesh Transfer — sparse barycentric correspondence matrix pre-computed per backend; runtime topology transfer is a single sparse matrix-vector product in O(V_h) time. 2. RBF Skeleton Fitting — Radial Basis Function regression with Kabsch rotation alignment yields the 77-joint identity-adapted skeleton transforms in a single linear solve per identity. 3. Linear Blend Skinning (LBS) — standard LBS with joint-orient (T-pose-relative) parameterization; GPU-accelerated via NVIDIA Warp custom kernels with torch.export-compatible interface.

Optional: shallow pose-dependent corrective MLP (2 hidden layers, ReLU activations) for surface artifact reduction.

This model was developed independently by NVIDIA.

Number of model parameters: - Core analytical pipeline: 0 learned parameters (closed-form) - Principal Component Analysis (PCA) for SOMA-shape model: 128 principal components x ~18,000 mid-LOD vertices x 3 = ~3.1 x 10^6 coefficients (pre-fitted, not gradient-trained) - Optional pose corrective Multilayer Perceptron (MLP) layers: ~1 × 10^8 parameters (if enabled)

Computational Load

Throughput: > 7,033 posed meshes/second on NVIDIA A100 80GB (batch size 128, GPU Warp path)
Latency: 2.1 ms per mesh (batch = 1, GPU); 12.1 ms (batch = 1, CPU 32-core)
Skeleton fitting: < 1.68 ms (batch = 1)
Training compute: N/A — core pipeline requires no gradient-based training; PCA shape space fitted offline from body scan data.

Input(s):

Input Type(s): Numerical tensors (floating-point)

Input Format(s): - Identity coefficients: floating-point tensor, shape (B, K) where K = 128 for SOMA-shape backend or backend-specific dimensionality for SMPL/SMPL-X/MHR/ANNY/Garment - Pose parameters: axis-angle vectors (B, 77, 3) or rotation matrices (B, 77, 3, 3) covering 77 articulated joints (excludes root dummy joint) - Optional root translation: (B, 3) in meters - Hand pose parameters: axis-angle (B, 25, 3) or rotation matrices (B, 25, 3, 3), with joint zero representing the wrist - Hand identity coefficients: (B, 20) for the native SOMA hand PCA, or the backend-specific MANO/MHR dimensionality

Input Parameters: One-Dimensional (1D) coefficient vectors; Three-Dimensional (3D) pose tensors

Other Properties Related to Input: - Identity coefficients should lie within the shape space of the respective backend (no hard clipping, but extreme out-of-distribution values may produce artifact geometry). - Pose parameters follow standard axis-angle convention; no clamping is applied. - All inputs are standard float32 PyTorch tensors. No pre-processing beyond normalization within each backend's identity model is required.

Output(s):

Output Type(s): Numerical tensors (3D geometry)

Output Format(s): PyTorch float32 tensors

Output Parameters: Three-Dimensional (3D)

  • Posed mesh vertices: (B, N_h, 3) where N_h depends on lod (mid: 18,056; low: 4,505; xlo: 612) -- world-space vertex positions in meters
  • Joint positions: (B, 77, 3) — world-space 3D joint positions in meters
  • Rest-shape vertices: (B, N_h, 3) in meters, with the same LOD-dependent N_h (intermediate output, available on request)
  • Hand vertices: (B, V_h, 3) in wrist-local coordinates, where V_h is 2,859 (mid), 718 (low), or 134 (xlo) per hand
  • Hand joints and transforms: (B, 25, 3) and (B, 25, 4, 4)

Other Properties Related to Output: All outputs are in meters. Body vertex count N_h is selected by SOMALayer(lod=...); low_lod=True is the legacy alias for lod="low". Joint count is fixed at 77.

Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated systems. By leveraging NVIDIA's hardware (GPU cores) and software frameworks (CUDA libraries, NVIDIA Warp), the model achieves real-time throughput exceeding 7,000 meshes per second at batch size 128 on an A100 GPU.

Software Integration:

Runtime Engine(s): * NVIDIA Warp (GPU-accelerated LBS kernel, torch.export-compatible)
* PyTorch (CPU and GPU fallback)
* N/A — No dependency on TAO, Riva, NeMo, or other NVIDIA SDK runtimes

Supported Hardware Microarchitecture Compatibility: * NVIDIA Ampere (A100, A30, A40, A10, RTX 3000-series) — tested on A100 80GB
* NVIDIA Hopper (H100) — forward compatible via Warp/CUDA
* NVIDIA Ada Lovelace (RTX 4000-series, L40)
* NVIDIA Turing (T4, RTX 2000-series)
* NVIDIA Volta (V100)
* Any NVIDIA GPU with CUDA support — model is lightweight and runs on any NVIDIA GPU
* CPU only (PyTorch fallback, no CUDA required)

Preferred/Supported Operating System(s): * Linux
* Windows (via PyTorch CPU/GPU path)

Model Version(s):

  • SOMA-X v0.3.0 — adds SOMAHandLayer for left/right hands at mid/low/xlo LODs, native hand identity and articulation-pose PCA models, and MANO interoperability using separately licensed user-supplied model files.
  • SOMA-X v0.2.3 — full-body release with SOMALayer and all six body identity backends.

Training, Testing, and Evaluation Datasets:

Training Dataset:

SOMA-shape Identity Model (Shape PCA): - SizeUSA — commercially licensed 3D body scan dataset; largest anthropometric survey of the U.S. population, covering diverse body shapes across age, sex, and BMI groups. Used to compute the 128-component PCA shape space for the SOMA-native identity backend. - TripleGangers — commercially licensed 3D body scan dataset purchased from TripleGangers, containing body scans of 303 individuals. Contributes additional shape diversity to the SOMA-shape PCA. - GarmentMeasurement PCA model — body shape data distilled from the GarmentMeasurement parametric model to augment the shape space with garment-relevant proportions.

Shallow MLP for Pose Correctives:

  • Bones RigPlay Dataset: 80,000 (pose,mesh) pairs samples from Bones RigPlay motion capture dataset owned by NVIDIA.

Data Modality - Other: 3D meshes and 3D motion data

Non-Audio, Image, Text Training Data Size: 20,000 3D meshes and 80,000 (pose,mesh) pairs.

Data Collection Method: Automatic/Sensors (structured-light 3D body scanners) for SizeUSA and TripleGangers; Synthetic for GarmentMeasurement distillation; motion capture for Bones RigPlay.
Labeling Method: Automatic/Sensors (body landmarks auto-detected from scans)

Dataset License(s): SizeUSA — commercially licensed (purchased by NVIDIA); TripleGangers — commercially licensed (purchased by NVIDIA); GarmentMeasurement — Generated using the source code; Bones RigPlay - commercially licensed (purchased by NVIDIA)

Testing Dataset:

Not applicable

Evaluation Dataset:

Not applicable

Data Collection Method: Synthetic
Labeling Method: Automatic
Dataset License(s): N/A

Inference:

Acceleration Engine: NVIDIA Warp (custom LBS kernel, torch.export-compatible); PyTorch (fallback)
Test Hardware: * NVIDIA A100 80GB (primary GPU benchmark hardware) * 32-core AMD EPYC 7763 (CPU benchmark hardware)

Ethical Considerations:

NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.

SOMA produces 3D human body meshes in a purely geometric and anonymous form; it does not process images, video, or any personally identifiable data at inference time. Input shape coefficients do not correspond to real individuals unless explicitly constructed to do so. Developers integrating SOMA into applications that may reconstruct or represent real individuals should ensure they have obtained appropriate consent and comply with applicable privacy regulations.

For more detailed information on ethical considerations for this model, please see the Model Card++ subcards: BIAS.md, EXPLAINABILITY.md, SAFETY_and_SECURITY.md, and PRIVACY.md.

Please report model quality, risk, or security vulnerabilities here.

Identity and Version

Repository
nvidia/SOMA-X
Publisher
NVIDIA
Task
Robotics
Modality
Control
Library
Not stated by the source
Parameters
Not stated by the source
Languages
Not stated by the source
Revision
104578ed58857f6faa7592fb83d0a2dad43c36fa
First published
2026-03-09
Last updated
2026-09-18

Files and Weights

33 files, 867.3 MB in total. The weights are 5 files totalling 796.3 MB in npz, pt.

Weights5 files · 796.3 MB
Configuration2 files · 26.5 KB
Documentation2 files · 22.4 KB
Other23 files · 70.9 MB
Repository1 file · 2.0 KB
Every file
FileTypeSizeSHA-256
MHR/mhr_model_lod1.ptWeights696.1 MB 352e271a6c42
MHR/mhr_model_lod6.ptWeights2.1 MB 28839b7ee0eb
SOMAHand.npzWeights993.8 KB 233e9225c1da
SOMA_neutral.npzWeights26.6 MB 5829af4a5333
correctives_model.ptWeights70.5 MB 411d061a6510
SOMA-X-HF-MANIFEST.jsonConfiguration6.5 KB
SOMA_procedural_transforms.jsonConfiguration20.0 KB
LICENSEDocumentation11.4 KB
README.mdDocumentation11.0 KB
Anny/SOMA_wrap.objOther2.6 MB
Anny/base_body.objOther1.2 MB
GarmentMeasurements/SOMA_wrap.objOther2.2 MB
GarmentMeasurements/mean.objOther3.6 MB
MANO/SOMA_wrap_left.objOther194.6 KB
MANO/SOMA_wrap_right.objOther194.6 KB
MANO/base_hand_left.objOther49.4 KB
MANO/base_hand_right.objOther49.4 KB
MHR/SOMA_wrap_lod1.objOther1.8 MB
MHR/base_body_lod1.objOther2.3 MB
MHR/base_body_lod6.objOther39.0 KB
SMPL/SOMA_wrap.objOther2.7 MB
SMPL/base_body.objOther457.4 KB
SMPL/smpl_anim.npyOther1.0 MB 1126c0b377f0
SMPLX/SOMA_wrap.objOther2.6 MB
SMPLX/base_body.objOther701.0 KB
SOMA_template_rig.usdaOther28.3 MB a952cb7bdf4c
example_animation.npyOther5.6 MB f424385dae31
images/banner.pngOther48.5 KB
images/mhr2soma.gifOther543.0 KB d3ea90a5e41d
images/smpl2soma.gifOther1.4 MB 8ec04d057a47
images/soma-in-action.gifOther8.4 MB 85290dc989a9
images/soma_correctives.gifOther4.8 MB d071c2042768
.gitattributesRepository2.0 KB

License and Download

License
apache-2.0
Access
Open weights, no gate
Download size
796.3 MB
Download from NVIDIA

Released by NVIDIA through its official repository on Hugging Face. Read the license.

Memory Requirements

PrecisionWeights in memory
As published796.3 MB

Weights only, from the published parameter count; the key-value cache and runtime add to this.

Questions About SOMA-X

Can I use SOMA-X commercially?

Yes. SOMA-X is released under Apache License 2.0. The Apache License 2.0 is a permissive open-source license. It permits commercial use, modification and redistribution. It requires keeping the license and copyright notices and any NOTICE file, stating significant changes, and it includes an express patent grant from contributors.

Similar Models

LIBERO 4in1(liberospatial / liberoobject / liberogoal / libero10)共 53.19 GB 的 Wan2.2-VAE 编码 latent 缓存: 训练 LIBERO policy(action head / VLA)时直接读取 latent 缓存,避免重复 VAE 编码。 - 窗口模式: windowed(--windowed) - 输出:.pt 文件,每 episode 一个 - dataset(推荐): https://huggingface.co/datasets/MangoGoes/libero4in1wan2.2vaelatentdataset - model(本仓库): https://huggingface.co/MangoGoes/libero4in1wan2.2vaelatentcosmosstyle

Open weights other cosmos

Model · Robotics

wam_ctxpool_bmethod

Hyeonmo Kang

Wan2.2-TI2V-5B video DiT + 48-joint action head, trainingmode=joint. The base is suhyeok's finalized B-method recipe: a teacher-forced (sigma=0.25) self-EMA teacher plus an iBOT prototype loss at L18 L18, gamma=0.01, two-view. On top of it the 3 PAST cond latent frames are pooled into one motion frame before a chosen block. These are NOT the surrogate ctxpool runs. The surrogate line (older base, pd8 x GA1) lives in hmkang/wamctxpoolxattn and hmkang/wamctxpoolavg. Do not compare across the two sets. Geometry: 4-latin (numframesin=25, numframesout=41, fdf 2) = 4 cond + 2 future latent slots, 96 tokens per latent frame, 576 tokens per row. Effective batch 16 clips x GA 2 x 2 views = 64 rows…

Open weights apache-2.0 wan2.2

weighted/imatrix quants of https://huggingface.co/DeepCybo/PhysBrain1.5-8B For a convenient overview and download list, visit our model page for this model. static quants are available at https://huggingface.co/mradermacher/PhysBrain1.5-8B-GGUF This is a vision model - mmproj files (if any) will be in the static repository. If you are unsure how to use GGUF files, refer to one of TheBloke's READMEs for more details, including on how to concatenate multi-part files. (sorted by size, not necessarily quality. IQ-quants are often preferable over similar sized non-IQ quants) Here is a handy graph by ikawrakow comparing some lower-quality quant And here are Artefact2's thoughts on the matter…

Open weights transformers

This repository contains checkpoints and evaluation artifacts for GR00T fine-tuning. Each epoch folder is a separate model checkpoint; the repository root is an index. - Same 50 total LIBERO Spatial trajectories for every version (5,971 frames). - Vision encoder, language model, and the full action head/DiT are trainable. - Eight epochs maximum; 125 optimizer updates per epoch. - Every epoch checkpoint is uploaded and hash-verified before local weight eviction. - Every checkpoint is evaluated on all ten Spatial tasks, with 50 fixed initial states per task: 500 rollouts. - Success-rate plots use completed simulator evaluations, not training losses. Checkpoints and evaluations appear as the…

Open weights

Model · Robotics

FastWAM-TDAA

Xizhou Bu

本仓保存 RoboTwin 实测模型、TDAA codec、原始结果和 4,000 个视频。 在 FastWAM 中接入预训练 TDAA version3bin24 编解码器,将 [32,14] 动作块编码为 [8,16] latent。策略在 latent 空间做 flow matching,再解码为 32 步绝对关节动作。decoder 使用任务向量和由已执行动作历史的 DCT24 特征生成的 phase;每个 episode 重置历史。 frozen 控制整个 TDAA codec,FastWAM 策略仍参与训练。联合训练额外加入动作重构与进度预测损失。动作 token 从 32 个变为 8 个,仅表示动作表示压缩;本次没有端到端加速测量。 - TDAA codec 与配置:codec.pt、config.json、metadata.json、datasetstatistics.json、taskembeddings.json,来自 80,000 步 AE。 代码仓的下载工具按固定版本获取文件并校验 SHA-256。在代码仓安装环境后运行: 权重保存到 checkpoints/released/{official,tdaafrozen}/step002725.pt。Wan VAE/T5/tokenizer 等基础模型、训练数据和 RoboTwin 仿真资产需另外准备,详见 GitHub 复现说明。本次不包含 optimizer/scheduler 完整训练状态。 以下为 FastWAM 基线与 TDAA Frozen=True 两组实测训练的共同设置,已与各自保存的…

Open weights

LeRobot policy checkpoints uploaded by goalgen/uploadhfcheckpoints.sh. Each subfolder contains the deployment-ready pretrainedmodel/ payload (model.safetensors + config.json + pre/postprocessor + trainconfig.json).

Open weights lerobot