Amin Rezaei

Waterloo, Canada Blockchain Engineer | CS M.Math. @ UWaterloo

Results 2 repositories owned by


                                            Amin Rezaei

memory-efficient-attention

121

Stars

12

Forks

Watchers

Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch

AminRezaei0x443

PyTorch-LIT

154

Stars

6

Forks

Watchers

Lite Inference Toolkit (LIT) for PyTorch

AminRezaei0x443