Skip to content
This repository has been archived by the owner on Sep 1, 2024. It is now read-only.

[Feature Request] Add option to use functorch for BasicEnsemble #162

Open
luisenp opened this issue Aug 5, 2022 · 2 comments
Open

[Feature Request] Add option to use functorch for BasicEnsemble #162

luisenp opened this issue Aug 5, 2022 · 2 comments
Labels
enhancement New feature or request good first issue Good for newcomers

Comments

@luisenp
Copy link
Contributor

luisenp commented Aug 5, 2022

🚀 Feature Request

Change BasicEnsemble to optionally use functorch.vmap.

Motivation and Pitch

Is your feature request related to a problem? Please describe.

BasicEnsemble lets the user provide arbitrary models, which are stacked together using a very naive loop-based implementation. We should be able to do this more efficiently now using functorch.
@luisenp luisenp added enhancement New feature or request good first issue Good for newcomers labels Aug 5, 2022
@natolambert
Copy link
Contributor

Torch has a vmap??? !!!

@luisenp
Copy link
Contributor Author

luisenp commented Aug 5, 2022 •
edited
Loading

It's via functorch which is a separate library but I think they work closely with PyTorch's team.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
enhancement New feature or request good first issue Good for newcomers
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@luisenp @natolambert