botorch icon indicating copy to clipboard operation
botorch copied to clipboard

Fix posterior method in `BatchedMultiOutputGPyTorchModel` for tracing JIT

Open SaiAakash opened this issue 1 year ago • 0 comments

Motivation

Fixes #2591. Generates the MTMVN for the independent task case slightly differently when jit traced.

Have you read the Contributing Guidelines on pull requests?

Yes

Test Plan

A unit test test_posterior_in_trace_mode has been added to test_gpytorch.py

Related PRs

NA

SaiAakash avatar Oct 23 '24 17:10 SaiAakash