This video shows how to setup a VST drum instrument within an Ableton Live drum rack and apply colours for Push and also how to rout multi-channel audio out of the plugin to dedicated channels in the rack for FX
Battery is used as an example but this will apply to any plugin you wish to use within a drum rack