Skip to content

v0.3.0

Latest
Compare
Choose a tag to compare
@fferflo fferflo released this 11 Jun 13:48
· 41 commits to master since this release

einx v0.3.0

Added

  • Add partial support for tinygrad.
    • Supported:
      • einx.rearrange
      • einx.{elementwise|add|multiply|where|...}
      • einx.{reduce|sum|mean|...}
      • einx.{vmap_with_axis|flip|softmax|...}
      • einx.dot
    • Not supported:
      • einx.vmap (no vmap in tinygrad)
      • einx.{index|get_at|set_at|...} (due to relying on einx.vmap)

Changed

  • Use tf.gather_nd instead of x[y] to implement einx.get_at for Tensorflow.

Fixed

  • Allow empty tuples and lists as constraints for ellipsis parameters.
  • Fix shorthand notation in einx.dot.