News for hackers. Stuff that matters.

A comment on Useful algorithms that are not optimized by Jax, PyTorch, or TensorFlow. Parent comment

Re:Useful algorithms that are not optimized by Jax, PyTorch,... by sillysaurusx on Thursday July 22, 2021 @10:20PM (#27926524)
This is entirely possible. You have direct access to XLA, after all: https://jax.readthedocs.io/en/latest/notebooks/XLA_in_Python...

And XLA has dynamic shape semantics (currently unused by jax) via SetDimensionSize: https://www.tensorflow.org/xla/operation_semantics#setdimens...

[ Parent | on HN ]

There are no comments on this comment yet.