This repository is a public repo that includes implementation of core layers commonly used in current machine learning models.
conda create -n ml_layers python=3.10
conda activate ml_layers
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia