Supported Model Formats
Focus
Focus
Prisma AIRS

Supported Model Formats

Table of Contents

Supported Model Formats

Prisma AIRS AI Model Security supports scanning and security checks across various AI and ML model formats.
Where Can I Use This?What Do I Need?
  • Prisma AIRS (Model Security)
  • Prisma AIRS Model Security License
AI Model Security checks are supported on the following formats:
Model File TypeFile ExtensionModel Description
Microsoft Cognitive ToolkitcntkModels saved in Microsoft Cognitive Toolkit format.
JAX/Flax ModelsflaxModels created with Flax, a neural network library for JAX.
GGUF ModelsggufGeneral-purpose model format using GGUF.
Keras 3.x archivekeras3Newer Keras models using the latest Keras version.
Keras 1.x/2.x archivekeras_legacyOlder Keras models, often saved with HDF5.
Keras pickled modelkeras_pickleKeras models saved with Python pickle.
Keras 1.x/2.x HDF5keras_legacy_h5Legacy Models in HDF5 format.
Keras 3.x HDF5keras3_h5Models in HDF5 format, compatible with legacy and newer versions.
Keras weights filekeras_weightsSeparate files storing only model weights.
Keras model JSONkeras_model_jsonModels saved in JSON format for architecture storage.
Keras metadata protobufkeras_metadataAuxiliary files that store metadata for Keras models.
LightGBM ModelslightgbmGradient boosting models using LightGBM.
Llamafile executablellamafileDistribute and Run LLMs with a single file.
Apache MXNet modelmxnetModels saved in Apache MXNet format.
Numpy Array FilesnumpyArrays saved in .npy format.
Numpy Zip Filesnumpy_zipArrays compressed in .npz format.
Numpy Pickle Filesnumpy_pickleArrays serialized with pickle.
ONNX ModelsonnxModels saved in Open Neural Network Exchange format.
OpenVINO Binary Weightsopenvino_binCompiled binary files for OpenVINO.
OpenVINO XML Graphopenvino_xmlXML files storing OpenVINO model metadata.
Pickle FilespickleModels serialized using Python's pickle.
PyTorch v0.1.1 tarpytorch_v0_1_1Models saved with v0.1.1 PyTorch version.
PyTorch v0.1.10 stackedpytorch_v0_1_10Models saved with v0.1.10 PyTorch version.
PyTorch v1.3+ zippytorch_v1_13Models saved with v1.3+ PyTorch version.
PyTorch TorchScriptpytorch_torch_scriptPyTorch's format for serializing models.
PyTorch model archivepytorch_archiveArchived files containing serialized models.
Rockchip RKNN modelrknnModels saved in Rockchip Neural Network (RKNN) format.
Safetensors WeightssafetensorsModels saved using safetensors format for secure tensor storage.
Safetensors Indexsafetensors_indexIndex files for safetensors.
SKLearn ModelssklearnScikit-learn models serialized for deployment.
TensorRT EnginetensorrtNVIDIA's TensorRT models optimized for inference.
TensorFlow SavedModeltensorflowTensorFlow's standard saved model format.
TensorFlow Hub moduletf_hubModels from TensorFlow Hub.
TensorFlow MetaGraphtf_meta_graphTensorFlow's MetaGraph format for exporting graphs.
TensorFlow Lite/LiteRTlitertTensorFlow Lightweight format.
TensorFlow Lite JSONlitert_jsonLightweight format for mobile and embedded devices.
TensorFlow.js modeltf_jsTensorFlow.js format for models running in the browser.
Torch7 ModelstorchGeneral format for PyTorch models.
JSON FilesjsonJSON-based configurations or model descriptions.
Joblib serializedjoblibPython library that facilitates efficient serialization and deserialization of Python objects.
Tar archivetarA single file that bundles multiple files and directories together, preserving their original structure, permissions, and other file system metadata.
Zip archivezipA file format that combines multiple files into a single compressed archive, making it easier to store, send, and share.
NVIDIA NeMo modelnemoScalable and cloud-native generative AI framework.
Zlib compressedzlibLossless data compression method.
Gzip compressedgzipLossless data compression and decompression.
Bzip2 compressedbzip2Files compressed using the bzip2 algorithm.
LZMA compressedlzmaLossless data compression algorithm with high compression ratio.
XZ compressedxzData compression tool in Linux.
LZ4 compressedlz4Extremely fast compression algorithm.
7-Zip archive7_zipCompressed archive file format created by the open-source 7-Zip tool.
Hydra confighydraOpen-source Python framework.
YAML fileyamlHuman-readable data serialization language, often used for configuration files, to structure data like lists and key-value pairs.