LICENSE
README.md
setup.py
mlax/__init__.py
mlax/_utils.py
mlax/module.py
mlax/nn/__init__.py
mlax/nn/batchnorm.py
mlax/nn/bias.py
mlax/nn/conv.py
mlax/nn/embed.py
mlax/nn/f.py
mlax/nn/functional.py
mlax/nn/linear.py
mlax/nn/parallel.py
mlax/nn/scaler.py
mlax/nn/series.py
mlax_nn.egg-info/PKG-INFO
mlax_nn.egg-info/SOURCES.txt
mlax_nn.egg-info/dependency_links.txt
mlax_nn.egg-info/requires.txt
mlax_nn.egg-info/top_level.txt