Open sxjscience opened 3 years ago
Add Performer to our model zoo:
Code in flax: https://github.com/google-research/google-research/tree/master/performer/fast_self_attention
Some reference implementations in PyTorch: https://github.com/lucidrains/performer-pytorch
One choice is to add a pytorch-based model.
Description
Add Performer to our model zoo:
Code in flax: https://github.com/google-research/google-research/tree/master/performer/fast_self_attention
Some reference implementations in PyTorch: https://github.com/lucidrains/performer-pytorch
One choice is to add a pytorch-based model.