search

facebookresearch / beanmachine

A library that allows for inference on probabilistic models
https://beanmachine.org/
MIT License
264 stars 49 forks source link

BMInference API to compile and run BM models from Beanstalk #1801

Open AishwaryaSivaraman opened 1 year ago

AishwaryaSivaraman commented 1 year ago

Summary: We currently have BMGInference, that takes a BM python program produces a valid BMG graph and runs inference on the BMG C++ backend. In the same vein, this diff implements a BMInference API that takes a BM python program, optimizes it (currently uses the same optimization path as BMG) and runs BM inference methods to compute the samples. This allows us to separate the type of optimizations each backend needs and allows us to test end-to-end with an inference method.

Differential Revision: D40853055

facebook-github-bot commented 1 year ago

This pull request was exported from Phabricator. Differential Revision: D40853055

facebook-github-bot commented 1 year ago

This pull request was exported from Phabricator. Differential Revision: D40853055