All modules for which code is available
- jax._src.lax.lax
- jax._src.numpy.fft
- jax._src.numpy.lax_numpy
- jax._src.numpy.linalg
- jax._src.ops.scatter
- jax._src.random
- jax._src.scipy.linalg
- jax._src.scipy.ndimage
- jax._src.scipy.signal
- jax._src.third_party.numpy.linalg
- jax.dtypes
- numpy
- symjax.base
- symjax.data.cifar10
- symjax.data.cifar100
- symjax.data.dsprites
- symjax.data.emnist
- symjax.data.fashionmnist
- symjax.data.mnist
- symjax.data.stl10
- symjax.data.svhn
- symjax.data.utils
- symjax.nn.initializers
- symjax.nn.layers
- symjax.nn.losses
- symjax.nn.ops_nn
- symjax.nn.optimizers
- symjax.nn.schedules
- symjax.probabilities.base
- symjax.rl.agents
- symjax.rl.ddpg
- symjax.rl.utils
- symjax.tensor.base
- symjax.tensor.interpolation
- symjax.tensor.linalg
- symjax.tensor.ops_numpy
- symjax.tensor.ops_special
- symjax.tensor.signal