This is a general repository for me to store my implementations from scratch of popular model architectures/cool papers I find while doing AI research. Currently, all of these implementations are all over the place outside this repo, but I'm planning on cleaning up and aggregating them here to help others who may be looking for implementations and to improve at JAX and C++.
Current implementations:
Name | C++ | Python | JAX |
---|---|---|---|
Transformer | link | ||
Linear Regression | link | ||
CNN | link | ||
Projected Gradient Descent | link | ||