We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Merge pull request #310 from superbobry:main PiperOrigin-RevId: 688907348
Merge pull request #79 from jax-ml:version PiperOrigin-RevId: 508682582
Merge pull request #78 from jax-ml:backwards-xla PiperOrigin-RevId: 508552584
Merge pull request #6 from jax-ml/multidevice Lazily load CUDA modules before launch