NNTikZ

A collection of TikZ diagrams for neural network and deep learning concepts. This repository was created to provide a set of open-source diagrams for use in academic papers and presentations.

If you use this repository in your research or project, please cite it as follows:

@misc{love2024nntikz,
    author = {Fraser Love},
    title = {NNTikZ - TikZ Diagrams for Deep Learning and Neural Networks},
    year = 2024,
    url = {https://github.com/fraserlove/nntikz},
    note = {GitHub repository}
}

Examples

Transformer

Multi-Head Attention

Neural Network

Attention Mechanism

Gated Recurrent Unit (GRU)

RNN Encoder-Decoder

Backpropagation Through Time (BPTT)