jax-ml

icon indicating a website link https://docs.jax.dev

icon location Pushing back the limits on numerical computing.

Results 1 repositories owned by jax-ml

oryx

194
Stars
9
Forks
Watchers

Oryx is a library for probabilistic programming and deep learning built on top of Jax.

verified publisher icon jax-ml