This is a complination of straightforward implementations for neural networks and associated algorithms. These implementations are thoroughly documented with accompanying explanations.
It is important to note that this project was developed solely for educational purposes, to gain a deeper understanding of algorithms. These implementations may not be suitable for real-world applications. It is highly recommended to utilize other statoe-of-the-art libraries for building models.
- Multi-head attention
- Transformer building blocks
PyTorch
- Center Crop,
- Random Crop
- Adjust Contrast
- Adjust Brightness
- Adjust saturation
JAX
- Center Crop,
- Random Crop
- Adjust Contrast
- Adjust Brightness
- Adjust saturation
- AutoEncoder
- Latent Diffusion Models
- Sampling algorithms for stable diffusion