flash_mla/__init__.py (5 lines of code) (raw):

__version__ = "1.0.0" from flash_mla.flash_mla_interface import ( get_mla_metadata, flash_mla_with_kvcache, )