Skip to content

Remove PyTorch JIT trace functionality - #457

Open
nikitakuklev wants to merge 3 commits into
xopt-org:mainfrom
nikitakuklev:remove-jit-functionality
Open

Remove PyTorch JIT trace functionality#457
nikitakuklev wants to merge 3 commits into
xopt-org:mainfrom
nikitakuklev:remove-jit-functionality

Conversation

@nikitakuklev

@nikitakuklev nikitakuklev commented Aug 20, 2026

Copy link
Copy Markdown
Collaborator

Removes the torch.jit.trace-based compilation paths for GP models and acquisition functions, keeping only torch.compile, since TorchScript is deprecated upstream. Also fixes a latent bug in the grad=True path of torch_compile_gp_model.

@codecov

codecov Bot commented Aug 20, 2026

Copy link
Copy Markdown

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant