NGC | Catalog
CatalogContainersMegaMolBART gPRC Service v0.1

MegaMolBART gPRC Service v0.1

Logo for MegaMolBART gPRC Service v0.1
A gRPC service to generate SMILES using MegaMolBART model.
Latest Tag
April 1, 2024
Compressed Size
5.94 GB
Multinode Support
Multi-Arch Support
0.1.2 (Latest) Security Scan Results

Linux / amd64

Sorry, your browser does not support inline SVG.

What Is MegaMolBART gRPC service?

MegaMolBART is a model trained on SMILES string and this container deploys MegaMolBART model for infrencing. The container bring up a gRPC service available in port 500051.

Getting Started

MegaMolBART gRPC service is currently used with Cheminformatics Demo application (also available as a container). Please refer to Cheminformatics resource for bringing up the complete suite.

Please following these steps to start MegaMolBART gRPC service in stand-alone mode.

ngc registry model download-version "nvidia/clara/megamolbart:0.1.2"
  • Start an instance of the Docker image using the following command:
# E.g. for Docker version 19.03 or later
docker run \
--gpus all \
--rm \
-v $(pwd)/megamolbart_v0.1/:/models/megamolbart \

# E.g. for Docker version 19.02 or later
docker run \
--runtime nvidia \
--rm \
-v $(pwd)/megamolbart_v0.1/:/models/megamolbart \


Protobuf definition of this service is available in Cheminformatics resource .

Please use the following command to generate language specific stubs.

python -m grpc_tools.protoc -I./grpc/ \
 --<>_out=generated \
 --experimental_allow_proto3_optional \
 --grpc_python_out=generated \

The service has three functions:

  • SmilesToEmbedding: Returns the latent space embedding of an input SMILES string.
  • FindSimilars: Generates SMILES using the model by probing the vicinity of input SMILES's position in latent space.
  • Interpolate: Generates SMILES using the model by probing the latent space between two input SMILES.
  • GetIteration: Returns the checkpoint iteration version of the deployed model.

Current usage


By pulling and using the container, you accept the terms and conditions of Apache 2.0 license