Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
Primary LanguagePythonMIT LicenseMIT