google-deepmind/penzai

Issues

• Defining `_repr_html_` for Pytree dataclasses which uses `pz.ts.render_to_html` leads to infinite loop

  #26 opened 23 days ago by femtomc
  3

• A few HTML feature requests...

  #21 opened 23 days ago by femtomc
  3

• Customized `selector` functions for user-defined `Pytree` classes

  #10 opened a month ago by femtomc
  3

• Synchronize sliders across array data in Pytree

  #13 opened a month ago by femtomc
  3

• importing model from other jax-based libraries for weight visualization

  #17 opened 2 months ago by Jconn
  1

• Penzai as a training framework

  #16 opened 2 months ago by rosinality
  1

• Keep axes in nmap

  #7 opened 2 months ago by srush
  5

• Feature request: Animate button

  #9 opened 2 months ago by srush
  0

• Default `pytree_dataclass` for inheritors of a common baseclass

  #11 opened 2 months ago by femtomc
  2

• Auto unwrap for full reductions.

  #8 opened 2 months ago by srush
  2