David Buterez

Hi, I am experimenting with the VLAE model in a graph setting ([VGAE](https://arxiv.org/abs/1611.07308)). The main change to the standard VLAE is a graph convolution-based encoder and a reconstruction loss term....

[Feature] Support for a scatter 'concatenate' or 'groupby' operation

9

Hi, thanks for the amazing work so far! I was wondering if it would be possible to efficiently support a scatter operation that instead of reducing (e.g. using sum, mean,...

[Feature Request] Sparse tensor support for `attn_bias` in `memory_efficient_attention`

1

# 🚀 Feature I think that it would be very helpful to allow sparse tensors to be used as masks (`attn_bias`) in the `memory_efficient_attention` function. ## Motivation As the documentation...