├── .gitignore ├── LICENSE.md ├── README.md ├── Taskfile.yml ├── app └── readme-apps │ └── main.go ├── async ├── async_test.go ├── command_pool.go ├── command_pool_test.go ├── dataloader.go ├── dataloader_test.go ├── docs.md ├── staging_pool.go └── staging_pool_test.go ├── cgo_bridge ├── TEST_README.md ├── advanced_test.go ├── bridge.go ├── bridge.h ├── bridge.m ├── bridge_device.h ├── bridge_device.m ├── bridge_graph.h ├── bridge_graph.m ├── bridge_inference.h ├── bridge_inference.m ├── bridge_memory.h ├── bridge_memory.m ├── bridge_optimizer.h ├── bridge_optimizer.m ├── bridge_test ├── bridge_test.go ├── bridge_training.h ├── bridge_training.m ├── bridge_types.h ├── build.go ├── docs.md ├── dynamic_engine_test.go ├── inference_test.go ├── optimizer_test.go ├── parameter_interpreter.c ├── parameter_interpreter.h ├── setup_test.go ├── sgd └── training_test.go ├── checkpoints ├── checkpoint.go ├── checkpoint_test.go ├── docs.md ├── onnx.go ├── onnx.pb.go └── onnx.proto ├── docs ├── DOCUMENTATION_PLAN.md ├── README.md ├── examples │ ├── activation-functions-demo.md │ ├── cats-dogs-classification.md │ ├── computer-vision-pipeline.md │ ├── house-price-regression.md │ └── inference-engine-demo.md ├── getting-started │ ├── README.md │ ├── basic-concepts.md │ ├── installation.md │ └── quick-start.md ├── guides │ ├── architecture.md │ ├── checkpoints.md │ ├── computer-vision.md │ ├── inference-engine.md │ ├── layers.md │ ├── loss-functions.md │ ├── memory-management.md │ ├── optimizers.md │ ├── performance.md │ ├── progress-tracking.md │ └── visualization.md └── tutorials │ ├── README.md │ ├── cnn-tutorial.md │ ├── custom-layers.md │ ├── mixed-precision.md │ ├── mlp-tutorial.md │ ├── onnx-integration.md │ └── regression-tutorial.md ├── engine ├── architecture_test.go ├── docs.md ├── engine_test.go ├── inference_engine.go ├── inference_engine_dynamic_test.go ├── inference_engine_test.go ├── model_engine.go ├── model_engine_test.go ├── normalization_test.go ├── setup_test.go ├── training_engine.go └── training_engine_test.go ├── enhancements.md ├── examples ├── README.md ├── basic-tensor-operations │ ├── README.md │ ├── basic-tensor-operations │ └── main.go ├── cnn-image-classification │ ├── README.md │ ├── cnn-image-classification │ ├── main.go │ └── output │ │ └── plots │ │ ├── confusion_matrix.json │ │ ├── epoch_2 │ │ ├── confusion_matrix.json │ │ └── training_curves.json │ │ ├── epoch_4 │ │ ├── confusion_matrix.json │ │ └── training_curves.json │ │ ├── learning_rate_schedule.json │ │ ├── precision_recall.json │ │ ├── roc_curve.json │ │ └── training_curves.json ├── complete-training-pipeline │ ├── README.md │ ├── complete-training-pipeline │ └── main.go ├── mixed-precision-demo │ ├── README.md │ ├── go.sum │ ├── main.go │ └── mixed-precision-demo ├── model-serialization │ ├── README.md │ ├── demo_model.json │ ├── demo_model.onnx │ ├── main.go │ └── model-serialization ├── sigmoid-demo │ ├── README.md │ ├── main.go │ ├── sigmoid-demo │ └── test_onnx.go ├── tanh-demo │ ├── README.md │ ├── main.go │ ├── tanh-demo │ └── test_onnx.go └── training-with-progress │ └── training_with_progress.go ├── go.mod ├── go.sum ├── layers ├── docs.md ├── layer.go ├── layer_test.go ├── sigmoid_test.go └── tanh_test.go ├── memory ├── docs.md ├── manager.go ├── memory_test.go ├── optimization │ ├── buffer_pool.go │ └── buffer_pool_test.go ├── tensor.go └── tensor_copy_test.go ├── optimizer ├── adadelta.go ├── adadelta_test.go ├── adagrad.go ├── adagrad_test.go ├── adam.go ├── adam_test.go ├── docs.md ├── interface.go ├── interface_test.go ├── lbfgs.go ├── lbfgs_test.go ├── nadam.go ├── nadam_test.go ├── optimizer.test ├── rmsprop.go ├── rmsprop_test.go ├── sgd.go ├── sgd_test.go ├── state_helpers.go └── state_helpers_test.go ├── training ├── checkpoint_integration.go ├── docs.md ├── labels.go ├── labels_test.go ├── metrics.go ├── metrics_test.go ├── model_inferencer.go ├── model_trainer.go ├── plotting_service.go ├── plotting_service_test.go ├── progress.go ├── progress_example.go ├── progress_test.go ├── scheduler.go ├── scheduler_test.go ├── sidecar_helper.go ├── simple_trainer.go ├── sparse_cross_entropy_test.go ├── visualization.go └── visualization_test.go └── vision ├── dataloader ├── cache.go ├── cache_test.go ├── dataloader.go ├── dataloader_test.go ├── docs.md ├── shared_cache.go └── shared_cache_test.go ├── dataset ├── cats_dogs.go ├── cats_dogs_test.go ├── docs.md ├── image_folder.go └── image_folder_test.go └── preprocessing ├── docs.md ├── image.go └── image_test.go /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/README.md -------------------------------------------------------------------------------- /Taskfile.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/Taskfile.yml -------------------------------------------------------------------------------- /app/readme-apps/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/app/readme-apps/main.go -------------------------------------------------------------------------------- /async/async_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/async/async_test.go -------------------------------------------------------------------------------- /async/command_pool.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/async/command_pool.go -------------------------------------------------------------------------------- /async/command_pool_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/async/command_pool_test.go -------------------------------------------------------------------------------- /async/dataloader.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/async/dataloader.go -------------------------------------------------------------------------------- /async/dataloader_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/async/dataloader_test.go -------------------------------------------------------------------------------- /async/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/async/docs.md -------------------------------------------------------------------------------- /async/staging_pool.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/async/staging_pool.go -------------------------------------------------------------------------------- /async/staging_pool_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/async/staging_pool_test.go -------------------------------------------------------------------------------- /cgo_bridge/TEST_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/TEST_README.md -------------------------------------------------------------------------------- /cgo_bridge/advanced_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/advanced_test.go -------------------------------------------------------------------------------- /cgo_bridge/bridge.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge.go -------------------------------------------------------------------------------- /cgo_bridge/bridge.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge.h -------------------------------------------------------------------------------- /cgo_bridge/bridge.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge.m -------------------------------------------------------------------------------- /cgo_bridge/bridge_device.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_device.h -------------------------------------------------------------------------------- /cgo_bridge/bridge_device.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_device.m -------------------------------------------------------------------------------- /cgo_bridge/bridge_graph.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_graph.h -------------------------------------------------------------------------------- /cgo_bridge/bridge_graph.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_graph.m -------------------------------------------------------------------------------- /cgo_bridge/bridge_inference.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_inference.h -------------------------------------------------------------------------------- /cgo_bridge/bridge_inference.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_inference.m -------------------------------------------------------------------------------- /cgo_bridge/bridge_memory.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_memory.h -------------------------------------------------------------------------------- /cgo_bridge/bridge_memory.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_memory.m -------------------------------------------------------------------------------- /cgo_bridge/bridge_optimizer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_optimizer.h -------------------------------------------------------------------------------- /cgo_bridge/bridge_optimizer.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_optimizer.m -------------------------------------------------------------------------------- /cgo_bridge/bridge_test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_test -------------------------------------------------------------------------------- /cgo_bridge/bridge_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_test.go -------------------------------------------------------------------------------- /cgo_bridge/bridge_training.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_training.h -------------------------------------------------------------------------------- /cgo_bridge/bridge_training.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_training.m -------------------------------------------------------------------------------- /cgo_bridge/bridge_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/bridge_types.h -------------------------------------------------------------------------------- /cgo_bridge/build.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/build.go -------------------------------------------------------------------------------- /cgo_bridge/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/docs.md -------------------------------------------------------------------------------- /cgo_bridge/dynamic_engine_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/dynamic_engine_test.go -------------------------------------------------------------------------------- /cgo_bridge/inference_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/inference_test.go -------------------------------------------------------------------------------- /cgo_bridge/optimizer_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/optimizer_test.go -------------------------------------------------------------------------------- /cgo_bridge/parameter_interpreter.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/parameter_interpreter.c -------------------------------------------------------------------------------- /cgo_bridge/parameter_interpreter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/parameter_interpreter.h -------------------------------------------------------------------------------- /cgo_bridge/setup_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/setup_test.go -------------------------------------------------------------------------------- /cgo_bridge/sgd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/sgd -------------------------------------------------------------------------------- /cgo_bridge/training_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/cgo_bridge/training_test.go -------------------------------------------------------------------------------- /checkpoints/checkpoint.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/checkpoints/checkpoint.go -------------------------------------------------------------------------------- /checkpoints/checkpoint_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/checkpoints/checkpoint_test.go -------------------------------------------------------------------------------- /checkpoints/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/checkpoints/docs.md -------------------------------------------------------------------------------- /checkpoints/onnx.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/checkpoints/onnx.go -------------------------------------------------------------------------------- /checkpoints/onnx.pb.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/checkpoints/onnx.pb.go -------------------------------------------------------------------------------- /checkpoints/onnx.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/checkpoints/onnx.proto -------------------------------------------------------------------------------- /docs/DOCUMENTATION_PLAN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/DOCUMENTATION_PLAN.md -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/examples/activation-functions-demo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/examples/activation-functions-demo.md -------------------------------------------------------------------------------- /docs/examples/cats-dogs-classification.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/examples/cats-dogs-classification.md -------------------------------------------------------------------------------- /docs/examples/computer-vision-pipeline.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/examples/computer-vision-pipeline.md -------------------------------------------------------------------------------- /docs/examples/house-price-regression.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/examples/house-price-regression.md -------------------------------------------------------------------------------- /docs/examples/inference-engine-demo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/examples/inference-engine-demo.md -------------------------------------------------------------------------------- /docs/getting-started/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/getting-started/README.md -------------------------------------------------------------------------------- /docs/getting-started/basic-concepts.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/getting-started/basic-concepts.md -------------------------------------------------------------------------------- /docs/getting-started/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/getting-started/installation.md -------------------------------------------------------------------------------- /docs/getting-started/quick-start.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/getting-started/quick-start.md -------------------------------------------------------------------------------- /docs/guides/architecture.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/architecture.md -------------------------------------------------------------------------------- /docs/guides/checkpoints.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/checkpoints.md -------------------------------------------------------------------------------- /docs/guides/computer-vision.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/computer-vision.md -------------------------------------------------------------------------------- /docs/guides/inference-engine.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/inference-engine.md -------------------------------------------------------------------------------- /docs/guides/layers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/layers.md -------------------------------------------------------------------------------- /docs/guides/loss-functions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/loss-functions.md -------------------------------------------------------------------------------- /docs/guides/memory-management.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/memory-management.md -------------------------------------------------------------------------------- /docs/guides/optimizers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/optimizers.md -------------------------------------------------------------------------------- /docs/guides/performance.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/performance.md -------------------------------------------------------------------------------- /docs/guides/progress-tracking.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/progress-tracking.md -------------------------------------------------------------------------------- /docs/guides/visualization.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/guides/visualization.md -------------------------------------------------------------------------------- /docs/tutorials/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/tutorials/README.md -------------------------------------------------------------------------------- /docs/tutorials/cnn-tutorial.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/tutorials/cnn-tutorial.md -------------------------------------------------------------------------------- /docs/tutorials/custom-layers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/tutorials/custom-layers.md -------------------------------------------------------------------------------- /docs/tutorials/mixed-precision.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/tutorials/mixed-precision.md -------------------------------------------------------------------------------- /docs/tutorials/mlp-tutorial.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/tutorials/mlp-tutorial.md -------------------------------------------------------------------------------- /docs/tutorials/onnx-integration.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/tutorials/onnx-integration.md -------------------------------------------------------------------------------- /docs/tutorials/regression-tutorial.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/docs/tutorials/regression-tutorial.md -------------------------------------------------------------------------------- /engine/architecture_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/architecture_test.go -------------------------------------------------------------------------------- /engine/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/docs.md -------------------------------------------------------------------------------- /engine/engine_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/engine_test.go -------------------------------------------------------------------------------- /engine/inference_engine.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/inference_engine.go -------------------------------------------------------------------------------- /engine/inference_engine_dynamic_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/inference_engine_dynamic_test.go -------------------------------------------------------------------------------- /engine/inference_engine_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/inference_engine_test.go -------------------------------------------------------------------------------- /engine/model_engine.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/model_engine.go -------------------------------------------------------------------------------- /engine/model_engine_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/model_engine_test.go -------------------------------------------------------------------------------- /engine/normalization_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/normalization_test.go -------------------------------------------------------------------------------- /engine/setup_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/setup_test.go -------------------------------------------------------------------------------- /engine/training_engine.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/training_engine.go -------------------------------------------------------------------------------- /engine/training_engine_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/engine/training_engine_test.go -------------------------------------------------------------------------------- /enhancements.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/enhancements.md -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/README.md -------------------------------------------------------------------------------- /examples/basic-tensor-operations/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/basic-tensor-operations/README.md -------------------------------------------------------------------------------- /examples/basic-tensor-operations/basic-tensor-operations: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/basic-tensor-operations/basic-tensor-operations -------------------------------------------------------------------------------- /examples/basic-tensor-operations/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/basic-tensor-operations/main.go -------------------------------------------------------------------------------- /examples/cnn-image-classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/README.md -------------------------------------------------------------------------------- /examples/cnn-image-classification/cnn-image-classification: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/cnn-image-classification -------------------------------------------------------------------------------- /examples/cnn-image-classification/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/main.go -------------------------------------------------------------------------------- /examples/cnn-image-classification/output/plots/confusion_matrix.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/output/plots/confusion_matrix.json -------------------------------------------------------------------------------- /examples/cnn-image-classification/output/plots/epoch_2/confusion_matrix.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/output/plots/epoch_2/confusion_matrix.json -------------------------------------------------------------------------------- /examples/cnn-image-classification/output/plots/epoch_2/training_curves.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/output/plots/epoch_2/training_curves.json -------------------------------------------------------------------------------- /examples/cnn-image-classification/output/plots/epoch_4/confusion_matrix.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/output/plots/epoch_4/confusion_matrix.json -------------------------------------------------------------------------------- /examples/cnn-image-classification/output/plots/epoch_4/training_curves.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/output/plots/epoch_4/training_curves.json -------------------------------------------------------------------------------- /examples/cnn-image-classification/output/plots/learning_rate_schedule.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/output/plots/learning_rate_schedule.json -------------------------------------------------------------------------------- /examples/cnn-image-classification/output/plots/precision_recall.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/output/plots/precision_recall.json -------------------------------------------------------------------------------- /examples/cnn-image-classification/output/plots/roc_curve.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/output/plots/roc_curve.json -------------------------------------------------------------------------------- /examples/cnn-image-classification/output/plots/training_curves.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/cnn-image-classification/output/plots/training_curves.json -------------------------------------------------------------------------------- /examples/complete-training-pipeline/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/complete-training-pipeline/README.md -------------------------------------------------------------------------------- /examples/complete-training-pipeline/complete-training-pipeline: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/complete-training-pipeline/complete-training-pipeline -------------------------------------------------------------------------------- /examples/complete-training-pipeline/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/complete-training-pipeline/main.go -------------------------------------------------------------------------------- /examples/mixed-precision-demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/mixed-precision-demo/README.md -------------------------------------------------------------------------------- /examples/mixed-precision-demo/go.sum: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/mixed-precision-demo/go.sum -------------------------------------------------------------------------------- /examples/mixed-precision-demo/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/mixed-precision-demo/main.go -------------------------------------------------------------------------------- /examples/mixed-precision-demo/mixed-precision-demo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/mixed-precision-demo/mixed-precision-demo -------------------------------------------------------------------------------- /examples/model-serialization/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/model-serialization/README.md -------------------------------------------------------------------------------- /examples/model-serialization/demo_model.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/model-serialization/demo_model.json -------------------------------------------------------------------------------- /examples/model-serialization/demo_model.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/model-serialization/demo_model.onnx -------------------------------------------------------------------------------- /examples/model-serialization/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/model-serialization/main.go -------------------------------------------------------------------------------- /examples/model-serialization/model-serialization: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/model-serialization/model-serialization -------------------------------------------------------------------------------- /examples/sigmoid-demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/sigmoid-demo/README.md -------------------------------------------------------------------------------- /examples/sigmoid-demo/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/sigmoid-demo/main.go -------------------------------------------------------------------------------- /examples/sigmoid-demo/sigmoid-demo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/sigmoid-demo/sigmoid-demo -------------------------------------------------------------------------------- /examples/sigmoid-demo/test_onnx.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/sigmoid-demo/test_onnx.go -------------------------------------------------------------------------------- /examples/tanh-demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/tanh-demo/README.md -------------------------------------------------------------------------------- /examples/tanh-demo/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/tanh-demo/main.go -------------------------------------------------------------------------------- /examples/tanh-demo/tanh-demo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/tanh-demo/tanh-demo -------------------------------------------------------------------------------- /examples/tanh-demo/test_onnx.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/tanh-demo/test_onnx.go -------------------------------------------------------------------------------- /examples/training-with-progress/training_with_progress.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/examples/training-with-progress/training_with_progress.go -------------------------------------------------------------------------------- /go.mod: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/go.mod -------------------------------------------------------------------------------- /go.sum: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/go.sum -------------------------------------------------------------------------------- /layers/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/layers/docs.md -------------------------------------------------------------------------------- /layers/layer.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/layers/layer.go -------------------------------------------------------------------------------- /layers/layer_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/layers/layer_test.go -------------------------------------------------------------------------------- /layers/sigmoid_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/layers/sigmoid_test.go -------------------------------------------------------------------------------- /layers/tanh_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/layers/tanh_test.go -------------------------------------------------------------------------------- /memory/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/memory/docs.md -------------------------------------------------------------------------------- /memory/manager.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/memory/manager.go -------------------------------------------------------------------------------- /memory/memory_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/memory/memory_test.go -------------------------------------------------------------------------------- /memory/optimization/buffer_pool.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/memory/optimization/buffer_pool.go -------------------------------------------------------------------------------- /memory/optimization/buffer_pool_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/memory/optimization/buffer_pool_test.go -------------------------------------------------------------------------------- /memory/tensor.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/memory/tensor.go -------------------------------------------------------------------------------- /memory/tensor_copy_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/memory/tensor_copy_test.go -------------------------------------------------------------------------------- /optimizer/adadelta.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/adadelta.go -------------------------------------------------------------------------------- /optimizer/adadelta_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/adadelta_test.go -------------------------------------------------------------------------------- /optimizer/adagrad.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/adagrad.go -------------------------------------------------------------------------------- /optimizer/adagrad_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/adagrad_test.go -------------------------------------------------------------------------------- /optimizer/adam.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/adam.go -------------------------------------------------------------------------------- /optimizer/adam_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/adam_test.go -------------------------------------------------------------------------------- /optimizer/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/docs.md -------------------------------------------------------------------------------- /optimizer/interface.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/interface.go -------------------------------------------------------------------------------- /optimizer/interface_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/interface_test.go -------------------------------------------------------------------------------- /optimizer/lbfgs.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/lbfgs.go -------------------------------------------------------------------------------- /optimizer/lbfgs_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/lbfgs_test.go -------------------------------------------------------------------------------- /optimizer/nadam.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/nadam.go -------------------------------------------------------------------------------- /optimizer/nadam_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/nadam_test.go -------------------------------------------------------------------------------- /optimizer/optimizer.test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/optimizer.test -------------------------------------------------------------------------------- /optimizer/rmsprop.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/rmsprop.go -------------------------------------------------------------------------------- /optimizer/rmsprop_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/rmsprop_test.go -------------------------------------------------------------------------------- /optimizer/sgd.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/sgd.go -------------------------------------------------------------------------------- /optimizer/sgd_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/sgd_test.go -------------------------------------------------------------------------------- /optimizer/state_helpers.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/state_helpers.go -------------------------------------------------------------------------------- /optimizer/state_helpers_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/optimizer/state_helpers_test.go -------------------------------------------------------------------------------- /training/checkpoint_integration.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/checkpoint_integration.go -------------------------------------------------------------------------------- /training/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/docs.md -------------------------------------------------------------------------------- /training/labels.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/labels.go -------------------------------------------------------------------------------- /training/labels_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/labels_test.go -------------------------------------------------------------------------------- /training/metrics.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/metrics.go -------------------------------------------------------------------------------- /training/metrics_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/metrics_test.go -------------------------------------------------------------------------------- /training/model_inferencer.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/model_inferencer.go -------------------------------------------------------------------------------- /training/model_trainer.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/model_trainer.go -------------------------------------------------------------------------------- /training/plotting_service.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/plotting_service.go -------------------------------------------------------------------------------- /training/plotting_service_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/plotting_service_test.go -------------------------------------------------------------------------------- /training/progress.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/progress.go -------------------------------------------------------------------------------- /training/progress_example.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/progress_example.go -------------------------------------------------------------------------------- /training/progress_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/progress_test.go -------------------------------------------------------------------------------- /training/scheduler.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/scheduler.go -------------------------------------------------------------------------------- /training/scheduler_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/scheduler_test.go -------------------------------------------------------------------------------- /training/sidecar_helper.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/sidecar_helper.go -------------------------------------------------------------------------------- /training/simple_trainer.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/simple_trainer.go -------------------------------------------------------------------------------- /training/sparse_cross_entropy_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/sparse_cross_entropy_test.go -------------------------------------------------------------------------------- /training/visualization.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/visualization.go -------------------------------------------------------------------------------- /training/visualization_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/training/visualization_test.go -------------------------------------------------------------------------------- /vision/dataloader/cache.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataloader/cache.go -------------------------------------------------------------------------------- /vision/dataloader/cache_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataloader/cache_test.go -------------------------------------------------------------------------------- /vision/dataloader/dataloader.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataloader/dataloader.go -------------------------------------------------------------------------------- /vision/dataloader/dataloader_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataloader/dataloader_test.go -------------------------------------------------------------------------------- /vision/dataloader/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataloader/docs.md -------------------------------------------------------------------------------- /vision/dataloader/shared_cache.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataloader/shared_cache.go -------------------------------------------------------------------------------- /vision/dataloader/shared_cache_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataloader/shared_cache_test.go -------------------------------------------------------------------------------- /vision/dataset/cats_dogs.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataset/cats_dogs.go -------------------------------------------------------------------------------- /vision/dataset/cats_dogs_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataset/cats_dogs_test.go -------------------------------------------------------------------------------- /vision/dataset/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataset/docs.md -------------------------------------------------------------------------------- /vision/dataset/image_folder.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataset/image_folder.go -------------------------------------------------------------------------------- /vision/dataset/image_folder_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/dataset/image_folder_test.go -------------------------------------------------------------------------------- /vision/preprocessing/docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/preprocessing/docs.md -------------------------------------------------------------------------------- /vision/preprocessing/image.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/preprocessing/image.go -------------------------------------------------------------------------------- /vision/preprocessing/image_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsawler/go-metal/HEAD/vision/preprocessing/image_test.go --------------------------------------------------------------------------------