Lovely Tensors => Lovely JAX
Build Lovely JAX to be as good as Lovely Tensors
Lovely Tensors is a Python Library (that I wrote) that extends PyTorch with the ability to better summarize and visualize tensors in a human-friendly way. It is particularly useful during development/experimentation in Jupyter Notebooks but is in no way limited to Notebooks.
JAX is the new shiny low-level accelerated numerical computing library that is set to overtake PyTorch in the AI research field. I’m working on adding a Lovely Tensors counterpart in JAX.
Grant Deliverables:
- Lovely JAX DeviceArray matches the functionality of Lovely Tensors
- String repr
- .rgb
- .chans
- .plot
- Lovely Tensors extended to visualize/summarize NN layers (Linear, Conv, BatchNorm, etc.)
Squad
Squad Lead: xl0
- Twitter @xl0xl0xl0
- Alexey Zaytsev#6786
- 0x45Fb23DFf8b6E73430f4e34688C820aA1134932F