alibool

`keep_ix / fg_probs.shape[1]` returns a float tensor which can not be used as index with torch 1.x.

add optimization subpackage in `functorch`

1

Jax has its optimization subpackage in `jax.scipy.optimize` but `functorch` does not, [this repo](https://github.com/rfeinman/pytorch-minimize) may provide some help.

make `minimize` jit-able?

1

Thanks for this great package, it seems that`minimize` is not jit-able? for example ``` from torchmin import minimize import torch from functools import partial def bar(y, x): return (x +...

Add video generation model[Feature]

2

### What's the feature? Add video generation model ### Any other context? _No response_