I actually prefer markdown based notes for math. The only thing I'm missing really in notion is linking from proofs to theorems/lemmas used in the proofs.