NeuralJAXwork: GPU Accelerated Lightweight ML Framework from Scratch with JAX
Primary LanguagePythonMIT LicenseMIT