Jamie Townsend

Check out the project I'm working on [here](https://github.com/j-towns/iprofiler). I managed to modify line_profiler to do what you're asking, by changing a few lines in _line_profiler.pyx (the modified version is in...

This would be useful for fastar. A note on the API: is it possible that most use cases are in the form of a fixed-point iteration? Gradient descent and the...

Surely `jax.named_scope` and `jax.named_call` ought to be mentioned [here](https://jax.readthedocs.io/en/latest/profiling.html#adding-custom-trace-events) in the docs? I'm happy to write a pr if helpful.

Ah thanks a lot for raising this. I've just replicated both of these issues on my own machine. I think a slightly more efficient way to resolve 1 would be...

@cyntsh are you familiar with asymmetric numeral systems (ANS)? This is a more recent and nowadays more widely used alternative to arithmetic coding, and on a high level the only...

@cyntsh how come you can't use the `Word` types and bit-wise operations defined in the Prelude? ([these ones](https://github.com/google-research/dex-lang/blob/4402e08e20994447bbce4b65f4e0ff863ca5ded8/lib/prelude.dx#L52-L57))

On the line ```python return tf.gradients(g, v, grad_ys=d_xs) ``` in @mattjj's code bove, you can see that we tell TF to take the grad of `g` w.r.t. `v`. I think...