Model reference · open weights

UNI2-h

Embeddings MahmoodLab Image embed 1 build Non-commercial 72k dl/mo

UNI2-h is an open-weight embedding model from MahmoodLab. UNI2-h (BF16) weighs 2.7 GB; the smallest configuration that runs it is RTX 3060 12 GB.

  • UNI2-h is a pretrained vision backbone developed by MahmoodLab for image feature extraction on histopathology images.
  • The model utilizes a custom ViT-H/14 architecture with 681 million parameters and was trained on over 200 million image tiles from H&E and IHC slides.
  • It is released under the CC-BY-NC-ND-4.0 license, restricting use to non-commercial academic research purposes.

Summary of the MahmoodLab/UNI2-h model card, 2026-10-01

What it is

Released byMahmoodLab
Released2025-01-10
VRAM2.7 GB for the weights

What it runs on

Memory and cards for UNI2-h (BF16)

2.7 GBweights, file size
1.1 GBruntime overhead
CardRunsMemory
RTX 3060 12 GBfits11.6 GB
RTX 4060 Ti 16 GBfits15.4 GB
RTX 3090 24 GBfits23.4 GB
RTX 4090 24 GBfits23.4 GB
RTX 5090 32 GBfits31.0 GB
L40S 48 GBfits44.0 GB
A100 80 GBfits78.2 GB
H100 80 GBfits78.1 GB
RTX PRO 6000 Blackwell 96 GBfits93.8 GB
DGX Spark (GB10) 128 GB unifiedfits107 GB
H200 141 GBfits138 GB
B200 180 GBfits176 GB

How it works

How embedding models work

Your textsentence / documentEncodermaps meaningVectorlist of numbersAn embedding model turns text into a vector, so similar meanings sit close together, which is the basis of search and RAG.
© 2026 AxForge · EU-hosted AI infrastructure Pricing Docs Trust Privacy Terms