Accelerate, Optimize performance with streamlined training and serving options with JAX.
Primary LanguagePythonApache License 2.0Apache-2.0