Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[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

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
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