Neural networks training pipeline based on PyTorch

My repo for training neural nets using pytorch-lightning and hydra

A lightweight, open-source, and intelligent wake word detection engine. Train custom, high-accuracy models with minimal effort.

Curriculum training

YOLOv12 Underwater Object Detection is an open-source suite for underwater object detection, built on YOLOv12. It offers an end-to-end pipeline with GPU-accelerated training, customizable data augmentations, real-time inference via Gradio, and support for model export (ONNX & PyTorch).

tracebloc notebook to launch and manage experiments in collaboration

Deep Learning training and deployment pipeline, reduce repetitive work from research to deployment

This repository features an image sharpening pipeline using Knowledge Distillation. A high-capacity Restormer acts as the teacher model, while a lightweight Mini-UNet is trained as the student to mimic its performance.

Training an image classification model with CIFAR-10 dataset

Train custom wake word models with openWakeWord. A granular 13-step pipeline with compatibility patches for torchaudio 2.10+, Piper TTS, and speechbrain. Generates tiny ONNX models (~200 KB) for real-time keyword detection — like building your own "Hey Siri" trigger. WSL2/Linux + CUDA required.

Immutable checkpoint storage for ML training pipelines. Kernel-level protection, anomaly detection, score-gated rollback, and self-healing recovery. Built in Rust.

🧠 Deep-Learning Evolution: Unified collection of TensorFlow & PyTorch projects, featuring custom CUDA kernels, distributed training, memory‑efficient methods, and production‑ready pipelines. Showcases advanced GPU optimizations, from foundational models to cutting‑edge architectures. 🚀

SPIRA Model Trainer v2 (redesigned pipeline) by @danlawand

Internship projects completed as part of the Shristi24 program offered by IIIT Hyderabad

Machine Learning in Production

AI Message Labels: Packaging and pipelines for deep learning text classification models

PyTorch detailed analysis to create Machine learning to Deep learning model

Configurable PyTorch training pipeline

YoloLint is a tool for automatic validation of dataset structure, annotation files, and image sizes in YOLO projects. It helps you catch typical errors in directory structure, YAML files, annotation files, and now also ensures all your images have the correct size before you start model training.

A Modular, Production-Style ML Pipeline with Class-Imbalance Handling