search

Blealtan / efficient-kan

An efficient pure-PyTorch implementation of Kolmogorov-Arnold Network (KAN).
MIT License
3.49k stars 306 forks source link

refactor: utilize Einops in appropriate areas to enhance readability #42

Open juvi21 opened 1 month ago

juvi21 commented 1 month ago

In certain areas, I thought the code would benefit from using Einops instead of standard PyTorch operations.