João Sousa Pinto

Results 16 comments of João Sousa Pinto
trafficstars

`jax.numpy.tan` call adds `chlo.tan` call to exported StableHLO; `stablehlo.tan` should be preferred

FWIW, ``` (.jax) ➜ jax git:(stablehlo.tan) python Python 3.12.4 (main, Jun 6 2024, 18:26:44) [Clang 15.0.0 (clang-1500.3.9.4)] on darwin Type "help", "copyright", "credits" or "license" for more information. >>> import...

Sparse matrix multiplication and output nse values

Thanks for confirming. I hope the example I provided motivates future work in this direction. :)

Sparse matrix multiplication and output nse values

FYI I added a prototype of what I needed [here](https://github.com/joaospinto/jax_alqp/blob/main/jax_alqp/sparse.py) and [here](https://github.com/joaospinto/jax_alqp/blob/main/jax_alqp/solver.py#L248). Given that in the dense case we require all shapes to be statically determined, it would not be...

Sparse matrix multiplication and output nse values

The approach I described above does run into one issue. Every matrix operation I do gets unrolled into JAX's expression graph, making it blow up in size. This makes JIT...

Break up Bazel example into a separate repo to actually show GitHub Actions working

Thanks for the pointers, @nicholasjng ! Very helpful.

Break up Bazel example into a separate repo to actually show GitHub Actions working

I got it to _mostly_ work on https://github.com/joaospinto/sip_python. Got one strange error on Windows + Python 3.13 (https://github.com/pypa/cibuildwheel/issues/2378).