Как утекать графические графические услуги решателей PDE в Python? ⇐ Python

[Anonymous]

Я работал над небольшим пакетом Python, чтобы решить класс PDE с использованием scipy.integrate.solve_ivp . Поскольку дискретизация становятся все более тонкими, время выполнения становится узким местом, особенно когда мне нужно решить PDE для большого количества различных начальных условий. Вот пример моей реализации в Google Colab. В записной книжке я также попытался использовать Cupy для передачи данных в GPU, выполнить шаг вперед, а затем перевести обратно в процессор, но передача передачи была слишком большой.

Подробнее здесь: https://stackoverflow.com/questions/794 ... -in-python