This 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
Categories
Data VisualizationLicense
MIT LicenseFollow NNlib.jl
Other Useful Business Software
Skillfully - The future of skills based hiring
Skillfully transforms hiring through AI-powered skill simulations that show you how candidates actually perform before you hire them. Our platform helps companies cut through AI-generated resumes and rehearsed interviews by validating real capabilities in action. Through dynamic job specific simulations and skill-based assessments, companies like Bloomberg and McKinsey have cut screening time by 50% while dramatically improving hire quality.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of NNlib.jl!