BasaltBasalt

autograd

attributes, graph, node, params, symbol, ops

nn

activations, initializers, loss, model, optim, tensor, layers

utils

bytes, collection, dataloader, datasets, perf_utils, rand_utils, tensorutils