LICENSE
README.md
pyproject.toml
src/DepthML/__init__.py
src/DepthML/block.py
src/DepthML/typing.py
src/DepthML.egg-info/PKG-INFO
src/DepthML.egg-info/SOURCES.txt
src/DepthML.egg-info/dependency_links.txt
src/DepthML.egg-info/requires.txt
src/DepthML.egg-info/top_level.txt
src/DepthML/activations/__init__.py
src/DepthML/activations/base_activation.py
src/DepthML/activations/common_activations.py
src/DepthML/initializers/__init__.py
src/DepthML/initializers/base_initializer.py
src/DepthML/initializers/glorot.py
src/DepthML/initializers/kaiming.py
src/DepthML/initializers/zeros.py
src/DepthML/layers/__init__.py
src/DepthML/layers/base_layer.py
src/DepthML/layers/conv2d.py
src/DepthML/layers/flatten.py
src/DepthML/layers/input.py
src/DepthML/layers/linear.py
src/DepthML/losses/__init__.py
src/DepthML/losses/base_loss.py
src/DepthML/losses/categorical_cross_entropy.py
src/DepthML/losses/mean_squared_error.py
src/DepthML/losses/softmax_categorical_cross_entropy.py
src/DepthML/models/__init__.py
src/DepthML/models/base_model.py
src/DepthML/models/sequential.py
src/DepthML/optimizers/__init__.py
src/DepthML/optimizers/base_optimizer.py
src/DepthML/optimizers/sgd.py
src/DepthML/utils/__init__.py
src/DepthML/utils/clip_grad_norm.py
src/DepthML/utils/get_xp.py