Welcome to ml4gw's documentation!


This documentation is a work in progress! If you have any questions or suggestions, please feel free to reach out to ml4gw@ligo.mit.edu


ml4gw is a library of pytorch utilities for training neural networks in service of gravitational wave physics applications.

The code can be found on github at https://github.com/ml4gw/ml4gw



Currently, the following projects are using ml4gw to support their research:

  • Aframe - Gravitational wave detection of binary black hole mergers

  • PE - Parameter estimation of gravitational wave signals


As this library is still very much a work in progress, we anticipate that novel use cases will encounter errors stemming from a lack of robustness. We encourage users who encounter these difficulties to file issues on GitHub, and we'll be happy to offer support to extend our coverage to new or improved functionality. We also strongly encourage ML users in the GW physics space to try their hand at working on these issues and joining on as collaborators! For more information about how to get involved, feel free to reach out to ml4gw@ligo.mit.edu. By bringing in new users with new use cases, we hope to develop this library into a truly general-purpose tool which makes DL more accessible for gravitational wave physicists everywhere.


We are grateful for the support of the U.S. National Science Foundation (NSF) Harnessing the Data Revolution (HDR) Institute for Accelerating AI Algorithms for Data Driven Discovery (A3D3) under Cooperative Agreement No. PHY-2117997.