Vmap TGLFNN-UKAEA ensemble instead of unrolling it. - #51

Vmap TGLFNN-UKAEA ensemble instead of unrolling it.#51

copybara-service[bot] merged 1 commit into

google-deepmind:maingoogle-deepmind/fusion_surrogates:mainfrom

theo-brown:tglfnn_vmap_ensembletheo-brown/fusion_surrogates:tglfnn_vmap_ensembleCopy head branch name to clipboard