DownloadThis package provides a library of functions useful for neural networks, such as softmax, sigmoid, batched multiplication, convolutions and pooling. Many of these are used by Flux.jl, which loads this package, but they may be used independently.
Features
- CUDA support
- AMDGPU support
- Automatic differentiation
- Defines gradients using ChainRules.jl
- GPU support is provided as package extensions
- Neural Network primitives with multiple backends
Project Samples
Categories
Data VisualizationLicense
MIT LicenseFollow NNlib.jl
You Might Also Like
HubSpot is an AI-powered customer platform with all the software, integrations, and resources you need to connect your marketing, sales, and customer service. HubSpot's connected platform enables you to grow your business faster by focusing on what matters most: your customers.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of NNlib.jl!