deepmodeling / deepmodeling/jax-fem

Would cuDSS be supported as a solver in the near future?

Open
#87 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
754
Forks
133
PR merge metrics
No merged PRs in 30d

Description

Would a cudss solver will be supported in the future? There has been efforts to create zero-copy interfaces between jax and cudss: https://github.com/johnviljoen/spineax. This will probably require replacing scipy csr matrices with jax's experimental sparse matrices. It does sound like a lot of work though!

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.