I took most math heavy notes in markdown with LaTeX. Sometimes, I used google sheets. Lately I've been using Observable[1].

[1] https://beta.observablehq.com/