Boltz ⚡

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

Installation

using Pkg
Pkg.add("Boltz")

Getting Started

using Boltz, Lux, Random
using JLD2 # Needed to load pretrained weights

model = Vision.AlexNet(; pretrained="ImageNet1K") # or "DEFAULT"
ps, st = Lux.setup(Random.default_rng(), model)

x = rand(Float32, 224, 224, 3, 1)
model(x, ps, Lux.testmode(st))