Borna Ahmadzadeh

Results 7 repositories owned by Borna Ahmadzadeh

neural-network-cuda

78
Stars
16
Forks
Watchers

Neural network from scratch in CUDA/C++

vit-pytorch

18
Stars
0
Forks
Watchers

PyTorch implementation of the vision transformer

attorch

523
Stars
28
Forks
Watchers

A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.

attention-in-vision

15
Stars
2
Forks
Watchers

PyTorch implementation of popular attention mechanisms in vision

flaim

19
Stars
1
Forks
Watchers

Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.

trap

99
Stars
6
Forks
Watchers

Autoregressive transformers in APL