LuxLib.jl

Backend for Lux.jl
Author LuxDL
Popularity
7 Stars
Updated Last
24 Days Ago
Started In
March 2023

LuxLib

GitHub Discussions Latest Docs Stable Docs

CI Buildkite Benchmarks codecov

Downloads Downloads

JET Testing Aqua QA ColPrac: Contributor's Guide on Collaborative Practices for Community Packages SciML Code Style

Backend for Lux.jl.

Tutorials

This is a developer-facing project and most users should not depend on it directly. As such, we don't have tutorials for this package. Instead, we recommend you check out the Lux tutorials.

What's the distinction from NNlib.jl?

This is currently a place to hold more specialized kernels and layer implementations for Lux.jl. Anyone is free to move these to NNlib.jl (this package is MIT licensed), but I probably don't have the time to do so myself. But incase you do, open an issue here and let me know I will delete the code from this package.

Changelog

Updating from v0.2 to v0.3

groupnorm with statistics tracking support has been removed.

Updating from v0.1 to v0.2

Support for CUDA has been moved to a weak dependency. If you want to use CUDA, you need to install and load LuxCUDA as using LuxCUDA or import LuxCUDA.