torch_sampling. Efficient reservoir sampling implementation for PyTorch
torchessian. Full loss Hessian spectrum approximation tool.