Meta OT Initializer

[bamos]

This is an initial version of a Meta OT initializer along with a notebook demonstrating it's usage.
Here is how the notebook renders in the website, and here are how the code docs render:

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[marcocuturi]

Thanks Brandon!! this looks fantastic as an addition!! a few minor comments + suggestion to use ent_reg_cost directly (open question would be to handle tau_... with MetaOT... Maybe for now set to 1...)

[codecov-commenter]

Codecov Report

Merging #145 (d27731a) into main (8a8e406) will increase coverage by 0.04%.
The diff coverage is 93.05%.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #145      +/-   ##
==========================================
+ Coverage   89.80%   89.84%   +0.04%     
==========================================
  Files          51       51              
  Lines        4971     5040      +69     
  Branches      505      509       +4     
==========================================
+ Hits         4464     4528      +64     
- Misses        386      389       +3     
- Partials      121      123       +2     

Impacted Files	Coverage Δ
ott/core/initializers.py	96.93% <93.05%> (-3.07%)	⬇️

[bamos]

@marcocuturi - Thanks, I just pulled the latest main, pushed in updates, responded inline/resolved the inline comments above, and re-ran the notebook with the latest version of the code

[marcocuturi]

This looks good for me!! @michalk8 want to take a look?

[michalk8]

Thanks a lot @bamos , I have made a few comments, will look at the notebook later!

[bamos]

Thanks @michalk8! I just pushed in most of your suggestions in bb34644, resolved them inline above, and requested a re-review. I've kept the last two outstanding ones open on the naming and geom.dtype. I'll re-run the notebook and push the new output there once these are resolved.

[michalk8]

Thanks @bamos , just minor changes needed.

[bamos]

I've updated the name to MetaInitializer and have addressed/responded to the other comments inline above. Should I also squash the commits here before merging/rebasing this in?

[michalk8]

Should I also squash the commits here before merging/rebasing this in?

There's no need to do this on your side, we do by default squash merges on PRs.

LGTM, thanks @bamos, I think this is a great addition!

[bamos]

Thanks for the help and for going through everything so thoroughly! :-)