I realized that some modules might not be needed anymore, like https://github.com/pesser/edflow/tree/dev/edflow/nn We should just remove them.