Anselm Levskaya

The app has a color picker for stroke/fill color selection. If you're not in the expanded desktop view perhaps you didn't see it?

@citron - so sorry for not noticing this comment earlier! My github notifications get swamped by my professional repos and I missed this. One probably confusing historical design decision for...

`make_causal_mask` accepts an input with an arbitrary number of batch dimensions, but a single terminal "length" position, as you said `[B..., L]` it then generates a `[B..., 1, L, L]`...

Hey - sorry for the delay. The existing utility fns are not meant to be in any way the "final word" on transformer APIs. One of the problems w. transformers...

Just noticing this issue for the first time... I've seen similarly weird issues with Mixins and Flax resolved in the past by simply changing the order of the multiple inheritance...

I'm going to close this since the changes in #2448 seem to address the main point, feel free to reopen.

Hey @jatentaki - so sorry for the delay: There's a lot of outstanding work 1) getting lifted versions of all the JAX AD machinery for Modules and/or 2) making some...

@jatentaki - sorry for the delay! @jheek - you may be interested in this too. I think I have a reasonable draft of a "lifted" version of odeint here in...

__NB:the below is probably completely obsolete now__ -- levskaya 2021.02.18 I was asked to post a utility function I use to delete DeviceArrays in colab: ```python import gc import jax...

@Jeevesh8 - there have been massive changes in the underlying JAX infrastructure since the above was written (incl more python logic moved into C++), I believe some new memory utilities...