LICENSE
MANIFEST.in
README.md
setup.py
lapjax/__init__.py
lapjax/__init__.pyi
lapjax/abstract_arrays.py
lapjax/abstract_arrays.pyi
lapjax/ad_checkpoint.py
lapjax/ad_checkpoint.pyi
lapjax/api_util.py
lapjax/api_util.pyi
lapjax/cloud_tpu_init.py
lapjax/cloud_tpu_init.pyi
lapjax/collect_profile.py
lapjax/collect_profile.pyi
lapjax/config.py
lapjax/config.pyi
lapjax/core.py
lapjax/core.pyi
lapjax/custom_batching.py
lapjax/custom_batching.pyi
lapjax/custom_derivatives.py
lapjax/custom_derivatives.pyi
lapjax/custom_transpose.py
lapjax/custom_transpose.pyi
lapjax/debug.py
lapjax/debug.pyi
lapjax/distributed.py
lapjax/distributed.pyi
lapjax/dlpack.py
lapjax/dlpack.pyi
lapjax/dtypes.py
lapjax/dtypes.pyi
lapjax/errors.py
lapjax/errors.pyi
lapjax/flatten_util.py
lapjax/flatten_util.pyi
lapjax/jaxpr_util.py
lapjax/jaxpr_util.pyi
lapjax/linear_util.py
lapjax/linear_util.pyi
lapjax/prng.py
lapjax/prng.pyi
lapjax/profiler.py
lapjax/profiler.pyi
lapjax/random.py
lapjax/random.pyi
lapjax/sharding.py
lapjax/sharding.pyi
lapjax/stages.py
lapjax/stages.pyi
lapjax/test_util.py
lapjax/test_util.pyi
lapjax/tree_util.py
lapjax/tree_util.pyi
lapjax/util.py
lapjax/util.pyi
lapjax/version.py
lapjax/version.pyi
lapjax.egg-info/PKG-INFO
lapjax.egg-info/SOURCES.txt
lapjax.egg-info/dependency_links.txt
lapjax.egg-info/requires.txt
lapjax.egg-info/top_level.txt
lapjax/_src/__init__.py
lapjax/_src/__init__.pyi
lapjax/_src/abstract_arrays.py
lapjax/_src/abstract_arrays.pyi
lapjax/_src/ad_checkpoint.py
lapjax/_src/ad_checkpoint.pyi
lapjax/_src/ad_util.py
lapjax/_src/ad_util.pyi
lapjax/_src/api.py
lapjax/_src/api.pyi
lapjax/_src/api_util.py
lapjax/_src/api_util.pyi
lapjax/_src/array.py
lapjax/_src/array.pyi
lapjax/_src/basearray.py
lapjax/_src/basearray.pyi
lapjax/_src/callback.py
lapjax/_src/callback.pyi
lapjax/_src/checkify.py
lapjax/_src/checkify.pyi
lapjax/_src/cloud_tpu_init.py
lapjax/_src/cloud_tpu_init.pyi
lapjax/_src/config.py
lapjax/_src/config.pyi
lapjax/_src/custom_api_util.py
lapjax/_src/custom_api_util.pyi
lapjax/_src/custom_batching.py
lapjax/_src/custom_batching.pyi
lapjax/_src/custom_derivatives.py
lapjax/_src/custom_derivatives.pyi
lapjax/_src/custom_transpose.py
lapjax/_src/custom_transpose.pyi
lapjax/_src/debugging.py
lapjax/_src/debugging.pyi
lapjax/_src/device_array.py
lapjax/_src/device_array.pyi
lapjax/_src/dispatch.py
lapjax/_src/dispatch.pyi
lapjax/_src/distributed.py
lapjax/_src/distributed.pyi
lapjax/_src/dlpack.py
lapjax/_src/dlpack.pyi
lapjax/_src/dtypes.py
lapjax/_src/dtypes.pyi
lapjax/_src/environment_info.py
lapjax/_src/environment_info.pyi
lapjax/_src/errors.py
lapjax/_src/errors.pyi
lapjax/_src/flatten_util.py
lapjax/_src/flatten_util.pyi
lapjax/_src/iree.py
lapjax/_src/iree.pyi
lapjax/_src/lax_reference.py
lapjax/_src/lax_reference.pyi
lapjax/_src/lazy_loader.py
lapjax/_src/lazy_loader.pyi
lapjax/_src/path.py
lapjax/_src/path.pyi
lapjax/_src/pretty_printer.py
lapjax/_src/pretty_printer.pyi
lapjax/_src/prng.py
lapjax/_src/prng.pyi
lapjax/_src/profiler.py
lapjax/_src/profiler.pyi
lapjax/_src/public_test_util.py
lapjax/_src/public_test_util.pyi
lapjax/_src/random.py
lapjax/_src/random.pyi
lapjax/_src/sharding.py
lapjax/_src/sharding.pyi
lapjax/_src/source_info_util.py
lapjax/_src/source_info_util.pyi
lapjax/_src/stages.py
lapjax/_src/stages.pyi
lapjax/_src/test_util.py
lapjax/_src/test_util.pyi
lapjax/_src/traceback_util.py
lapjax/_src/traceback_util.pyi
lapjax/_src/tree_util.py
lapjax/_src/tree_util.pyi
lapjax/_src/typing.py
lapjax/_src/typing.pyi
lapjax/_src/util.py
lapjax/_src/util.pyi
lapjax/_src/clusters/__init__.py
lapjax/_src/clusters/__init__.pyi
lapjax/_src/clusters/cloud_tpu_cluster.py
lapjax/_src/clusters/cloud_tpu_cluster.pyi
lapjax/_src/clusters/cluster.py
lapjax/_src/clusters/cluster.pyi
lapjax/_src/clusters/slurm_cluster.py
lapjax/_src/clusters/slurm_cluster.pyi
lapjax/_src/debugger/__init__.py
lapjax/_src/debugger/__init__.pyi
lapjax/_src/debugger/cli_debugger.py
lapjax/_src/debugger/cli_debugger.pyi
lapjax/_src/debugger/colab_debugger.py
lapjax/_src/debugger/colab_debugger.pyi
lapjax/_src/debugger/colab_lib.py
lapjax/_src/debugger/colab_lib.pyi
lapjax/_src/debugger/core.py
lapjax/_src/debugger/core.pyi
lapjax/_src/debugger/web_debugger.py
lapjax/_src/debugger/web_debugger.pyi
lapjax/_src/image/__init__.py
lapjax/_src/image/__init__.pyi
lapjax/_src/image/scale.py
lapjax/_src/image/scale.pyi
lapjax/_src/lax/__init__.py
lapjax/_src/lax/__init__.pyi
lapjax/_src/lax/ann.py
lapjax/_src/lax/ann.pyi
lapjax/_src/lax/convolution.py
lapjax/_src/lax/convolution.pyi
lapjax/_src/lax/eigh.py
lapjax/_src/lax/eigh.pyi
lapjax/_src/lax/fft.py
lapjax/_src/lax/fft.pyi
lapjax/_src/lax/lax.py
lapjax/_src/lax/lax.pyi
lapjax/_src/lax/linalg.py
lapjax/_src/lax/linalg.pyi
lapjax/_src/lax/other.py
lapjax/_src/lax/other.pyi
lapjax/_src/lax/parallel.py
lapjax/_src/lax/parallel.pyi
lapjax/_src/lax/qdwh.py
lapjax/_src/lax/qdwh.pyi
lapjax/_src/lax/slicing.py
lapjax/_src/lax/slicing.pyi
lapjax/_src/lax/stack.py
lapjax/_src/lax/stack.pyi
lapjax/_src/lax/svd.py
lapjax/_src/lax/svd.pyi
lapjax/_src/lax/utils.py
lapjax/_src/lax/utils.pyi
lapjax/_src/lax/windowed_reductions.py
lapjax/_src/lax/windowed_reductions.pyi
lapjax/_src/lax/control_flow/__init__.py
lapjax/_src/lax/control_flow/__init__.pyi
lapjax/_src/lax/control_flow/common.py
lapjax/_src/lax/control_flow/common.pyi
lapjax/_src/lax/control_flow/conditionals.py
lapjax/_src/lax/control_flow/conditionals.pyi
lapjax/_src/lax/control_flow/for_loop.py
lapjax/_src/lax/control_flow/for_loop.pyi
lapjax/_src/lax/control_flow/loops.py
lapjax/_src/lax/control_flow/loops.pyi
lapjax/_src/lax/control_flow/solves.py
lapjax/_src/lax/control_flow/solves.pyi
lapjax/_src/lib/__init__.py
lapjax/_src/lib/__init__.pyi
lapjax/_src/lib/xla_bridge.py
lapjax/_src/lib/xla_bridge.pyi
lapjax/_src/lib/mlir/__init__.py
lapjax/_src/lib/mlir/__init__.pyi
lapjax/_src/lib/mlir/dialects/__init__.py
lapjax/_src/lib/mlir/dialects/__init__.pyi
lapjax/_src/nn/__init__.py
lapjax/_src/nn/__init__.pyi
lapjax/_src/nn/functions.py
lapjax/_src/nn/functions.pyi
lapjax/_src/nn/initializers.py
lapjax/_src/nn/initializers.pyi
lapjax/_src/numpy/__init__.py
lapjax/_src/numpy/__init__.pyi
lapjax/_src/numpy/fft.py
lapjax/_src/numpy/fft.pyi
lapjax/_src/numpy/index_tricks.py
lapjax/_src/numpy/index_tricks.pyi
lapjax/_src/numpy/lax_numpy.py
lapjax/_src/numpy/lax_numpy.pyi
lapjax/_src/numpy/linalg.py
lapjax/_src/numpy/linalg.pyi
lapjax/_src/numpy/ndarray.py
lapjax/_src/numpy/ndarray.pyi
lapjax/_src/numpy/polynomial.py
lapjax/_src/numpy/polynomial.pyi
lapjax/_src/numpy/reductions.py
lapjax/_src/numpy/reductions.pyi
lapjax/_src/numpy/setops.py
lapjax/_src/numpy/setops.pyi
lapjax/_src/numpy/ufuncs.py
lapjax/_src/numpy/ufuncs.pyi
lapjax/_src/numpy/util.py
lapjax/_src/numpy/util.pyi
lapjax/_src/numpy/vectorize.py
lapjax/_src/numpy/vectorize.pyi
lapjax/_src/ops/__init__.py
lapjax/_src/ops/__init__.pyi
lapjax/_src/ops/scatter.py
lapjax/_src/ops/scatter.pyi
lapjax/_src/ops/special.py
lapjax/_src/ops/special.pyi
lapjax/_src/scipy/__init__.py
lapjax/_src/scipy/__init__.pyi
lapjax/_src/scipy/fft.py
lapjax/_src/scipy/fft.pyi
lapjax/_src/scipy/linalg.py
lapjax/_src/scipy/linalg.pyi
lapjax/_src/scipy/ndimage.py
lapjax/_src/scipy/ndimage.pyi
lapjax/_src/scipy/signal.py
lapjax/_src/scipy/signal.pyi
lapjax/_src/scipy/special.py
lapjax/_src/scipy/special.pyi
lapjax/_src/scipy/cluster/__init__.py
lapjax/_src/scipy/cluster/__init__.pyi
lapjax/_src/scipy/cluster/vq.py
lapjax/_src/scipy/cluster/vq.pyi
lapjax/_src/scipy/interpolate/__init__.py
lapjax/_src/scipy/interpolate/__init__.pyi
lapjax/_src/scipy/optimize/__init__.py
lapjax/_src/scipy/optimize/__init__.pyi
lapjax/_src/scipy/optimize/_lbfgs.py
lapjax/_src/scipy/optimize/_lbfgs.pyi
lapjax/_src/scipy/optimize/bfgs.py
lapjax/_src/scipy/optimize/bfgs.pyi
lapjax/_src/scipy/optimize/line_search.py
lapjax/_src/scipy/optimize/line_search.pyi
lapjax/_src/scipy/optimize/minimize.py
lapjax/_src/scipy/optimize/minimize.pyi
lapjax/_src/scipy/sparse/__init__.py
lapjax/_src/scipy/sparse/__init__.pyi
lapjax/_src/scipy/sparse/linalg.py
lapjax/_src/scipy/sparse/linalg.pyi
lapjax/_src/scipy/stats/__init__.py
lapjax/_src/scipy/stats/__init__.pyi
lapjax/_src/scipy/stats/_core.py
lapjax/_src/scipy/stats/_core.pyi
lapjax/_src/scipy/stats/bernoulli.py
lapjax/_src/scipy/stats/bernoulli.pyi
lapjax/_src/scipy/stats/beta.py
lapjax/_src/scipy/stats/beta.pyi
lapjax/_src/scipy/stats/betabinom.py
lapjax/_src/scipy/stats/betabinom.pyi
lapjax/_src/scipy/stats/cauchy.py
lapjax/_src/scipy/stats/cauchy.pyi
lapjax/_src/scipy/stats/chi2.py
lapjax/_src/scipy/stats/chi2.pyi
lapjax/_src/scipy/stats/dirichlet.py
lapjax/_src/scipy/stats/dirichlet.pyi
lapjax/_src/scipy/stats/expon.py
lapjax/_src/scipy/stats/expon.pyi
lapjax/_src/scipy/stats/gamma.py
lapjax/_src/scipy/stats/gamma.pyi
lapjax/_src/scipy/stats/gennorm.py
lapjax/_src/scipy/stats/gennorm.pyi
lapjax/_src/scipy/stats/geom.py
lapjax/_src/scipy/stats/geom.pyi
lapjax/_src/scipy/stats/kde.py
lapjax/_src/scipy/stats/kde.pyi
lapjax/_src/scipy/stats/laplace.py
lapjax/_src/scipy/stats/laplace.pyi
lapjax/_src/scipy/stats/logistic.py
lapjax/_src/scipy/stats/logistic.pyi
lapjax/_src/scipy/stats/multinomial.py
lapjax/_src/scipy/stats/multinomial.pyi
lapjax/_src/scipy/stats/multivariate_normal.py
lapjax/_src/scipy/stats/multivariate_normal.pyi
lapjax/_src/scipy/stats/nbinom.py
lapjax/_src/scipy/stats/nbinom.pyi
lapjax/_src/scipy/stats/norm.py
lapjax/_src/scipy/stats/norm.pyi
lapjax/_src/scipy/stats/pareto.py
lapjax/_src/scipy/stats/pareto.pyi
lapjax/_src/scipy/stats/poisson.py
lapjax/_src/scipy/stats/poisson.pyi
lapjax/_src/scipy/stats/t.py
lapjax/_src/scipy/stats/t.pyi
lapjax/_src/scipy/stats/truncnorm.py
lapjax/_src/scipy/stats/truncnorm.pyi
lapjax/_src/scipy/stats/uniform.py
lapjax/_src/scipy/stats/uniform.pyi
lapjax/_src/state/__init__.py
lapjax/_src/state/__init__.pyi
lapjax/_src/state/discharge.py
lapjax/_src/state/discharge.pyi
lapjax/_src/state/primitives.py
lapjax/_src/state/primitives.pyi
lapjax/_src/state/types.py
lapjax/_src/state/types.pyi
lapjax/_src/third_party/__init__.py
lapjax/_src/third_party/__init__.pyi
lapjax/_src/third_party/numpy/__init__.py
lapjax/_src/third_party/numpy/__init__.pyi
lapjax/_src/third_party/numpy/linalg.py
lapjax/_src/third_party/numpy/linalg.pyi
lapjax/_src/third_party/scipy/__init__.py
lapjax/_src/third_party/scipy/__init__.pyi
lapjax/_src/third_party/scipy/interpolate.py
lapjax/_src/third_party/scipy/interpolate.pyi
lapjax/_src/third_party/scipy/linalg.py
lapjax/_src/third_party/scipy/linalg.pyi
lapjax/_src/third_party/scipy/signal_helper.py
lapjax/_src/third_party/scipy/signal_helper.pyi
lapjax/example_libraries/__init__.py
lapjax/example_libraries/__init__.pyi
lapjax/example_libraries/optimizers.py
lapjax/example_libraries/optimizers.pyi
lapjax/example_libraries/stax.py
lapjax/example_libraries/stax.pyi
lapjax/experimental/__init__.py
lapjax/experimental/__init__.pyi
lapjax/experimental/callback.py
lapjax/experimental/callback.pyi
lapjax/experimental/checkify.py
lapjax/experimental/checkify.pyi
lapjax/experimental/custom_partitioning.py
lapjax/experimental/custom_partitioning.pyi
lapjax/experimental/global_device_array.py
lapjax/experimental/global_device_array.pyi
lapjax/experimental/host_callback.py
lapjax/experimental/host_callback.pyi
lapjax/experimental/jet.py
lapjax/experimental/jet.pyi
lapjax/experimental/maps.py
lapjax/experimental/maps.pyi
lapjax/experimental/mesh_utils.py
lapjax/experimental/mesh_utils.pyi
lapjax/experimental/multihost_utils.py
lapjax/experimental/multihost_utils.pyi
lapjax/experimental/ode.py
lapjax/experimental/ode.pyi
lapjax/experimental/pjit.py
lapjax/experimental/pjit.pyi
lapjax/experimental/x64_context.py
lapjax/experimental/x64_context.pyi
lapjax/experimental/compilation_cache/__init__.py
lapjax/experimental/compilation_cache/__init__.pyi
lapjax/experimental/compilation_cache/cache_interface.py
lapjax/experimental/compilation_cache/cache_interface.pyi
lapjax/experimental/compilation_cache/compilation_cache.py
lapjax/experimental/compilation_cache/compilation_cache.pyi
lapjax/experimental/compilation_cache/gfile_cache.py
lapjax/experimental/compilation_cache/gfile_cache.pyi
lapjax/experimental/gda_serialization/__init__.py
lapjax/experimental/gda_serialization/__init__.pyi
lapjax/experimental/gda_serialization/serialization.py
lapjax/experimental/gda_serialization/serialization.pyi
lapjax/experimental/gda_serialization/serialization_test.py
lapjax/experimental/gda_serialization/serialization_test.pyi
lapjax/experimental/jax2tf/__init__.py
lapjax/experimental/jax2tf/__init__.pyi
lapjax/experimental/jax2tf/call_tf.py
lapjax/experimental/jax2tf/call_tf.pyi
lapjax/experimental/jax2tf/impl_no_xla.py
lapjax/experimental/jax2tf/impl_no_xla.pyi
lapjax/experimental/jax2tf/jax2tf.py
lapjax/experimental/jax2tf/jax2tf.pyi
lapjax/experimental/jax2tf/shape_poly.py
lapjax/experimental/jax2tf/shape_poly.pyi
lapjax/experimental/jax2tf/examples/__init__.py
lapjax/experimental/jax2tf/examples/__init__.pyi
lapjax/experimental/jax2tf/examples/keras_reuse_main.py
lapjax/experimental/jax2tf/examples/keras_reuse_main.pyi
lapjax/experimental/jax2tf/examples/keras_reuse_main_test.py
lapjax/experimental/jax2tf/examples/keras_reuse_main_test.pyi
lapjax/experimental/jax2tf/examples/mnist_lib.py
lapjax/experimental/jax2tf/examples/mnist_lib.pyi
lapjax/experimental/jax2tf/examples/saved_model_lib.py
lapjax/experimental/jax2tf/examples/saved_model_lib.pyi
lapjax/experimental/jax2tf/examples/saved_model_main.py
lapjax/experimental/jax2tf/examples/saved_model_main.pyi
lapjax/experimental/jax2tf/examples/saved_model_main_test.py
lapjax/experimental/jax2tf/examples/saved_model_main_test.pyi
lapjax/experimental/jax2tf/examples/serving/__init__.py
lapjax/experimental/jax2tf/examples/serving/__init__.pyi
lapjax/experimental/jax2tf/examples/serving/model_server_request.py
lapjax/experimental/jax2tf/examples/serving/model_server_request.pyi
lapjax/experimental/jax2tf/tests/__init__.py
lapjax/experimental/jax2tf/tests/__init__.pyi
lapjax/experimental/jax2tf/tests/call_tf_test.py
lapjax/experimental/jax2tf/tests/call_tf_test.pyi
lapjax/experimental/jax2tf/tests/control_flow_ops_test.py
lapjax/experimental/jax2tf/tests/control_flow_ops_test.pyi
lapjax/experimental/jax2tf/tests/converters.py
lapjax/experimental/jax2tf/tests/converters.pyi
lapjax/experimental/jax2tf/tests/jax2tf_limitations.py
lapjax/experimental/jax2tf/tests/jax2tf_limitations.pyi
lapjax/experimental/jax2tf/tests/jax2tf_test.py
lapjax/experimental/jax2tf/tests/jax2tf_test.pyi
lapjax/experimental/jax2tf/tests/jax_primitives_coverage_test.py
lapjax/experimental/jax2tf/tests/jax_primitives_coverage_test.pyi
lapjax/experimental/jax2tf/tests/model_harness.py
lapjax/experimental/jax2tf/tests/model_harness.pyi
lapjax/experimental/jax2tf/tests/models_test_main.py
lapjax/experimental/jax2tf/tests/models_test_main.pyi
lapjax/experimental/jax2tf/tests/primitive_harness.py
lapjax/experimental/jax2tf/tests/primitive_harness.pyi
lapjax/experimental/jax2tf/tests/primitives_test.py
lapjax/experimental/jax2tf/tests/primitives_test.pyi
lapjax/experimental/jax2tf/tests/savedmodel_test.py
lapjax/experimental/jax2tf/tests/savedmodel_test.pyi
lapjax/experimental/jax2tf/tests/shape_poly_test.py
lapjax/experimental/jax2tf/tests/shape_poly_test.pyi
lapjax/experimental/jax2tf/tests/sharding_test.py
lapjax/experimental/jax2tf/tests/sharding_test.pyi
lapjax/experimental/jax2tf/tests/tf_test_util.py
lapjax/experimental/jax2tf/tests/tf_test_util.pyi
lapjax/experimental/sparse/__init__.py
lapjax/experimental/sparse/__init__.pyi
lapjax/experimental/sparse/_base.py
lapjax/experimental/sparse/_base.pyi
lapjax/experimental/sparse/ad.py
lapjax/experimental/sparse/ad.pyi
lapjax/experimental/sparse/api.py
lapjax/experimental/sparse/api.pyi
lapjax/experimental/sparse/bcoo.py
lapjax/experimental/sparse/bcoo.pyi
lapjax/experimental/sparse/bcsr.py
lapjax/experimental/sparse/bcsr.pyi
lapjax/experimental/sparse/coo.py
lapjax/experimental/sparse/coo.pyi
lapjax/experimental/sparse/csr.py
lapjax/experimental/sparse/csr.pyi
lapjax/experimental/sparse/linalg.py
lapjax/experimental/sparse/linalg.pyi
lapjax/experimental/sparse/random.py
lapjax/experimental/sparse/random.pyi
lapjax/experimental/sparse/transform.py
lapjax/experimental/sparse/transform.pyi
lapjax/experimental/sparse/util.py
lapjax/experimental/sparse/util.pyi
lapjax/image/__init__.py
lapjax/image/__init__.pyi
lapjax/interpreters/__init__.py
lapjax/interpreters/__init__.pyi
lapjax/interpreters/ad.py
lapjax/interpreters/ad.pyi
lapjax/interpreters/batching.py
lapjax/interpreters/batching.pyi
lapjax/interpreters/mlir.py
lapjax/interpreters/mlir.pyi
lapjax/interpreters/partial_eval.py
lapjax/interpreters/partial_eval.pyi
lapjax/interpreters/pxla.py
lapjax/interpreters/pxla.pyi
lapjax/interpreters/xla.py
lapjax/interpreters/xla.pyi
lapjax/lapsrc/__init__.py
lapjax/lapsrc/axis_utils.py
lapjax/lapsrc/check_utils.py
lapjax/lapsrc/func_utils.py
lapjax/lapsrc/function_class.py
lapjax/lapsrc/functions.py
lapjax/lapsrc/lapconfig.py
lapjax/lapsrc/laptuple.py
lapjax/lapsrc/laputils.py
lapjax/lapsrc/sparsinfo.py
lapjax/lapsrc/sparsutils.py
lapjax/lapsrc/wrap_list.py
lapjax/lapsrc/wrapper.py
lapjax/lax/__init__.py
lapjax/lax/__init__.pyi
lapjax/lax/linalg.py
lapjax/lax/linalg.pyi
lapjax/lib/__init__.py
lapjax/lib/__init__.pyi
lapjax/lib/xla_bridge.py
lapjax/lib/xla_bridge.pyi
lapjax/nn/__init__.py
lapjax/nn/__init__.pyi
lapjax/nn/initializers.py
lapjax/nn/initializers.pyi
lapjax/numpy/__init__.py
lapjax/numpy/__init__.pyi
lapjax/numpy/fft.py
lapjax/numpy/fft.pyi
lapjax/numpy/linalg.py
lapjax/numpy/linalg.pyi
lapjax/ops/__init__.py
lapjax/ops/__init__.pyi
lapjax/scipy/__init__.py
lapjax/scipy/__init__.pyi
lapjax/scipy/fft.py
lapjax/scipy/fft.pyi
lapjax/scipy/linalg.py
lapjax/scipy/linalg.pyi
lapjax/scipy/ndimage.py
lapjax/scipy/ndimage.pyi
lapjax/scipy/signal.py
lapjax/scipy/signal.pyi
lapjax/scipy/special.py
lapjax/scipy/special.pyi
lapjax/scipy/cluster/__init__.py
lapjax/scipy/cluster/__init__.pyi
lapjax/scipy/cluster/vq.py
lapjax/scipy/cluster/vq.pyi
lapjax/scipy/interpolate/__init__.py
lapjax/scipy/interpolate/__init__.pyi
lapjax/scipy/optimize/__init__.py
lapjax/scipy/optimize/__init__.pyi
lapjax/scipy/sparse/__init__.py
lapjax/scipy/sparse/__init__.pyi
lapjax/scipy/sparse/linalg.py
lapjax/scipy/sparse/linalg.pyi
lapjax/scipy/stats/__init__.py
lapjax/scipy/stats/__init__.pyi
lapjax/scipy/stats/bernoulli.py
lapjax/scipy/stats/bernoulli.pyi
lapjax/scipy/stats/beta.py
lapjax/scipy/stats/beta.pyi
lapjax/scipy/stats/betabinom.py
lapjax/scipy/stats/betabinom.pyi
lapjax/scipy/stats/cauchy.py
lapjax/scipy/stats/cauchy.pyi
lapjax/scipy/stats/chi2.py
lapjax/scipy/stats/chi2.pyi
lapjax/scipy/stats/dirichlet.py
lapjax/scipy/stats/dirichlet.pyi
lapjax/scipy/stats/expon.py
lapjax/scipy/stats/expon.pyi
lapjax/scipy/stats/gamma.py
lapjax/scipy/stats/gamma.pyi
lapjax/scipy/stats/gennorm.py
lapjax/scipy/stats/gennorm.pyi
lapjax/scipy/stats/geom.py
lapjax/scipy/stats/geom.pyi
lapjax/scipy/stats/laplace.py
lapjax/scipy/stats/laplace.pyi
lapjax/scipy/stats/logistic.py
lapjax/scipy/stats/logistic.pyi
lapjax/scipy/stats/multinomial.py
lapjax/scipy/stats/multinomial.pyi
lapjax/scipy/stats/multivariate_normal.py
lapjax/scipy/stats/multivariate_normal.pyi
lapjax/scipy/stats/nbinom.py
lapjax/scipy/stats/nbinom.pyi
lapjax/scipy/stats/norm.py
lapjax/scipy/stats/norm.pyi
lapjax/scipy/stats/pareto.py
lapjax/scipy/stats/pareto.pyi
lapjax/scipy/stats/poisson.py
lapjax/scipy/stats/poisson.pyi
lapjax/scipy/stats/t.py
lapjax/scipy/stats/t.pyi
lapjax/scipy/stats/truncnorm.py
lapjax/scipy/stats/truncnorm.pyi
lapjax/scipy/stats/uniform.py
lapjax/scipy/stats/uniform.pyi
lapjax/tools/__init__.py
lapjax/tools/__init__.pyi
lapjax/tools/colab_tpu.py
lapjax/tools/colab_tpu.pyi
lapjax/tools/jax_to_ir.py
lapjax/tools/jax_to_ir.pyi
test/test_funcs.py