⚝ One Hat Cyber Team ⚝
  • Dir : ~/usr/local/lib/python3.10/dist-packages/jax/_src/
    • Name Size Permission Action
    __pycache__ drwxr-xr-x Rename Delete
    clusters drwxr-xr-x Rename Delete
    cudnn drwxr-xr-x Rename Delete
    debugger drwxr-xr-x Rename Delete
    export drwxr-xr-x Rename Delete
    extend drwxr-xr-x Rename Delete
    image drwxr-xr-x Rename Delete
    internal_test_util drwxr-xr-x Rename Delete
    interpreters drwxr-xr-x Rename Delete
    lax drwxr-xr-x Rename Delete
    lib drwxr-xr-x Rename Delete
    nn drwxr-xr-x Rename Delete
    numpy drwxr-xr-x Rename Delete
    ops drwxr-xr-x Rename Delete
    pallas drwxr-xr-x Rename Delete
    scipy drwxr-xr-x Rename Delete
    state drwxr-xr-x Rename Delete
    third_party drwxr-xr-x Rename Delete
    __init__.py 0.567 KB -rw-r--r-- Edit Rename Delete
    abstract_arrays.py 3.513 KB -rw-r--r-- Edit Rename Delete
    ad_checkpoint.py 38.036 KB -rw-r--r-- Edit Rename Delete
    ad_util.py 4.066 KB -rw-r--r-- Edit Rename Delete
    api.py 118.14 KB -rw-r--r-- Edit Rename Delete
    api_util.py 27.62 KB -rw-r--r-- Edit Rename Delete
    array.py 45.548 KB -rw-r--r-- Edit Rename Delete
    basearray.py 6.351 KB -rw-r--r-- Edit Rename Delete
    basearray.pyi 12.717 KB -rw-r--r-- Edit Rename Delete
    blocked_sampler.py 5.533 KB -rw-r--r-- Edit Rename Delete
    cache_key.py 12.581 KB -rw-r--r-- Edit Rename Delete
    callback.py 23.231 KB -rw-r--r-- Edit Rename Delete
    checkify.py 54.864 KB -rw-r--r-- Edit Rename Delete
    cloud_tpu_init.py 3.437 KB -rw-r--r-- Edit Rename Delete
    compilation_cache.py 10.963 KB -rw-r--r-- Edit Rename Delete
    compilation_cache_interface.py 0.866 KB -rw-r--r-- Edit Rename Delete
    compiler.py 27.144 KB -rw-r--r-- Edit Rename Delete
    compute_on.py 2.025 KB -rw-r--r-- Edit Rename Delete
    config.py 65.105 KB -rw-r--r-- Edit Rename Delete
    core.py 107.222 KB -rw-r--r-- Edit Rename Delete
    custom_api_util.py 0.855 KB -rw-r--r-- Edit Rename Delete
    custom_batching.py 13.952 KB -rw-r--r-- Edit Rename Delete
    custom_derivatives.py 67.214 KB -rw-r--r-- Edit Rename Delete
    custom_partitioning.py 22.191 KB -rw-r--r-- Edit Rename Delete
    custom_partitioning_sharding_rule.py 15.517 KB -rw-r--r-- Edit Rename Delete
    custom_transpose.py 7.605 KB -rw-r--r-- Edit Rename Delete
    debugging.py 25.297 KB -rw-r--r-- Edit Rename Delete
    deprecations.py 4.596 KB -rw-r--r-- Edit Rename Delete
    dispatch.py 22.421 KB -rw-r--r-- Edit Rename Delete
    distributed.py 11.6 KB -rw-r--r-- Edit Rename Delete
    dlpack.py 11.273 KB -rw-r--r-- Edit Rename Delete
    dtypes.py 33.064 KB -rw-r--r-- Edit Rename Delete
    earray.py 4.395 KB -rw-r--r-- Edit Rename Delete
    effects.py 4.572 KB -rw-r--r-- Edit Rename Delete
    environment_info.py 1.829 KB -rw-r--r-- Edit Rename Delete
    errors.py 24.161 KB -rw-r--r-- Edit Rename Delete
    flatten_util.py 3.471 KB -rw-r--r-- Edit Rename Delete
    hardware_utils.py 1.688 KB -rw-r--r-- Edit Rename Delete
    jaxpr_util.py 6.288 KB -rw-r--r-- Edit Rename Delete
    lax_reference.py 17.737 KB -rw-r--r-- Edit Rename Delete
    layout.py 5.249 KB -rw-r--r-- Edit Rename Delete
    lazy_loader.py 1.799 KB -rw-r--r-- Edit Rename Delete
    linear_util.py 12.973 KB -rw-r--r-- Edit Rename Delete
    logging_config.py 4.348 KB -rw-r--r-- Edit Rename Delete
    lru_cache.py 6.505 KB -rw-r--r-- Edit Rename Delete
    mesh.py 16.534 KB -rw-r--r-- Edit Rename Delete
    mesh_utils.py 31.635 KB -rw-r--r-- Edit Rename Delete
    monitoring.py 3.797 KB -rw-r--r-- Edit Rename Delete
    op_shardings.py 3.803 KB -rw-r--r-- Edit Rename Delete
    partition_spec.py 2.135 KB -rw-r--r-- Edit Rename Delete
    path.py 1.418 KB -rw-r--r-- Edit Rename Delete
    pickle_util.py 2.258 KB -rw-r--r-- Edit Rename Delete
    pjit.py 122.143 KB -rw-r--r-- Edit Rename Delete
    pretty_printer.py 15.137 KB -rw-r--r-- Edit Rename Delete
    prng.py 43.312 KB -rw-r--r-- Edit Rename Delete
    profiler.py 15.815 KB -rw-r--r-- Edit Rename Delete
    public_test_util.py 10.891 KB -rw-r--r-- Edit Rename Delete
    random.py 97.925 KB -rw-r--r-- Edit Rename Delete
    shard_alike.py 3.934 KB -rw-r--r-- Edit Rename Delete
    sharding.py 7.912 KB -rw-r--r-- Edit Rename Delete
    sharding_impls.py 64.68 KB -rw-r--r-- Edit Rename Delete
    sharding_specs.py 8.443 KB -rw-r--r-- Edit Rename Delete
    source_info_util.py 9.902 KB -rw-r--r-- Edit Rename Delete
    sourcemap.py 6.484 KB -rw-r--r-- Edit Rename Delete
    stages.py 28.449 KB -rw-r--r-- Edit Rename Delete
    test_util.py 74.141 KB -rw-r--r-- Edit Rename Delete
    tpu_custom_call.py 25.508 KB -rw-r--r-- Edit Rename Delete
    traceback_util.py 8.89 KB -rw-r--r-- Edit Rename Delete
    tree.py 12.297 KB -rw-r--r-- Edit Rename Delete
    tree_util.py 46.22 KB -rw-r--r-- Edit Rename Delete
    typing.py 3.21 KB -rw-r--r-- Edit Rename Delete
    util.py 19.504 KB -rw-r--r-- Edit Rename Delete
    xla_bridge.py 45.294 KB -rw-r--r-- Edit Rename Delete
    xla_metadata.py 1.483 KB -rw-r--r-- Edit Rename Delete