Welcome to MLAX’s documentation!

Contents:

• Overview
  • Why MLAX?
  • Installation
  • Worked Examples
• Modules and Parameters
• Neural Network Layers
• Frequently Asked Questions
  • Modules’ __call__ function is throwing an error when jit-compiled.
  • Modules` filter / partition functions are throwing AttributeError: cannot filter/partition an uninitialized module
  • How to perform mixed-precision training in mlax?
• mlax
  • mlax package

Indices and tables

• Index

• Module Index

• Search Page