https://github.com/kyflores/ml-ks

Machine Learning Kitchen Sink
https://github.com/kyflores/ml-ks

Last synced: 18 days ago
JSON representation

Machine Learning Kitchen Sink

• Host: GitHub
• URL: https://github.com/kyflores/ml-ks
• Owner: kyflores
• License: mit
• Created: 2024-05-25T23:53:21.000Z (8 months ago)
• Default Branch: main
• Last Pushed: 2025-01-01T10:39:31.000Z (22 days ago)
• Last Synced: 2025-01-01T11:28:17.325Z (22 days ago)
• Language: Jupyter Notebook
• Size: 611 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# ml-ks

Machine Learning Kitchen Sink

## Mamba/Conda environment

This config assumes a CUDA system. When installing on ROCm you'll

need different pytorch packages, and possibly a different deepspeed.

```

mamba create -n mlks \

    python=3.11 \

    pip \

    pytorch \

    torchvision \

    torchaudio \

    pytorch-cuda=12.1 \

    transformers \

    datasets \

    accelerate \

    deepspeed \

    tqdm \

    torchinfo \

    kornia \

    jupyterlab \

    ipywidgets \

    onnx \

    -c pytorch -c nvidia -c conda-forge

```