deepmodeling / deepmodeling/jax-fem
Would cuDSS be supported as a solver in the near future?
Open
- 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.