Specify command line arguments for Python modules in distributed way (like abseil does)

Question

How can I specify command line arguments on a per module base, that are then parsed when the application is run. This is basically how ABSEIL flags work, but I would prefer a more lightweight solution. Can this be achieved with argparse or any other python library? Example:

# trainer.py
specify_argument('learning_rate', float)
...

# model.py
specify_argument('num_layers', int)
...

# main.py
def main(): 
  parse_args()
  ...

Thanks!