Lux.jl

API Reference

• LuxAMDGPU
  • Index
  • API
• LuxCUDA
  • Index
  • API Reference
• LuxDeviceUtils
  • Index
  • Preferences
  • Data Transfer
  • Miscellaneous
• LuxCore
  • Index
  • Abstract Types
  • General
  • Parameters
  • States
• LuxLib
  • Index
  • Dropout
  • Normalization
• WeightInitializers
  • Index
  • API Reference
• Boltz
  • Index
• Computer Vision Models
  • Classification Models: Native Lux Models
  • Building Blocks
    • Non-Public API
  • Classification Models: Imported from Metalhead.jl
    • Preprocessing
• Experimental Features
  • Index
  • Training
  • Parameter Freezing
  • Map over Layer
  • Debugging Functionality
  • Tied Parameters
  • Stateful Layer
  • Compact Layer API
• Flux Models to Lux Models
  • Index
  • Functions
  • Layers
• Built-In Layers
  • Index
  • Containers
  • Convolutional Layers
  • Dropout Layers
  • Pooling Layers
  • Recurrent Layers
  • Linear Layers
  • Misc. Helper Layers
  • Normalization Layers
  • Upsampling
• Utilities
  • Index
  • Device Management / Data Transfer
  • Weight Initialization
  • Miscellaneous Utilities
  • Updating Floating Point Precision
  • Truncated Stacktraces
• LuxTestUtils
  • Index
  • Testing using JET.jl
  • Gradient Correctness
• API Reference

This site is open source. Improve this page.