1. 13 4月, 2023 1 次提交
    • Ricardo Vieira's avatar
      Implement JAX dot with sparse constants · 3b425ec7
      Ricardo Vieira 提交于
      Non-constant sparse inputs can't be handled because JAX does not allow Scipy sparse matrices as inputs. We could implement a BCOO type explicitly but this would be JAX exclusive, and the user would need to use it from the get go, meaning such graphs would not be compatible with other backends.
      3b425ec7
  2. 07 4月, 2023 3 次提交
  3. 05 4月, 2023 5 次提交
  4. 31 3月, 2023 2 次提交
  5. 27 3月, 2023 1 次提交
  6. 23 3月, 2023 3 次提交
  7. 22 3月, 2023 6 次提交
  8. 21 3月, 2023 1 次提交
  9. 20 3月, 2023 1 次提交
  10. 17 3月, 2023 1 次提交
    • Ricardo Vieira's avatar
      Fix JAX erfinv failing test · 651ed9f4
      Ricardo Vieira 提交于
      JAX behavior at the boundary changed and no longer returns np.inf like scipy does. Fixed by evaluating at a non-boundary value.
      651ed9f4
  11. 11 3月, 2023 5 次提交
  12. 10 3月, 2023 1 次提交
  13. 08 3月, 2023 2 次提交
  14. 06 3月, 2023 2 次提交
  15. 25 2月, 2023 3 次提交
  16. 20 2月, 2023 1 次提交
  17. 14 2月, 2023 1 次提交
  18. 12 2月, 2023 1 次提交