Boltz ⚡

Accelerate ⚡ your ML research using pre-built Deep Learning Models with Lux.

Installation

using Pkg
Pkg.add("Boltz")

using Boltz, Lux, Metalhead

model, ps, st = resnet(:alexnet; pretrained=true)

CUDA is not loaded by default. To use GPUs follow Lux.jl documentation.

We have moved some dependencies into weak dependencies. This means that you will have to manually load them for certain features to be available.