search

ihmeuw-msca / regmod

General regression models.
BSD 2-Clause "Simplified" License
5 stars 3 forks source link

Bugfix/jaxmemory #60

Closed kels271828 closed 1 year ago

kels271828 commented 1 year ago

Fixed problem with JAX memory by moving JAX functions outside of class.