I like writing in a Jupyter notebook. You can use Markdown, write equations with Python and use a module like SymPy to transform to LaTeX, create graphs, etc.