Ownership is measured by how many commits each author made to each file in the current tree. It highlights bus-factor risk (files only one person has ever touched) and where knowledge is concentrated. The bus factor is the smallest number of people who are, between them, primary owner of more than half of the files.
Files only one author has ever changed. The more a file has changed, the more knowledge is at risk if that person leaves.
| File | Sole owner | Commits |
|---|---|---|
| tests/numerics/numerics_test_util.py | Peter Hawkins | 15 |
| docs/internals/constants.md | George Necula | 11 |
| jax/_src/internal_test_util/export_back_compat_test_data/export_with_specified_sharding.py | George Necula | 11 |
| jax/_src/internal_test_util/export_back_compat_test_data/export_with_unspecified_sharding.py | George Necula | 11 |
| tests/numerics/numerics_test_util_test.py | Peter Hawkins | 10 |
| docs/jax.experimental.sparse.rst | Jake VanderPlas | 8 |
| jax/experimental/key_reuse/__init__.py | Jake VanderPlas | 8 |
| tests/filecheck/jax_mlir_ext.filecheck.py | Peter Hawkins | 8 |
| docs/301/custom-derivatives.md | Matthew Johnson | 8 |
| jaxlib/config_test.py | Peter Hawkins | 7 |
| File | Sole owner | Commits |
|---|---|---|
| tests/numerics/numerics_test_util.py | Peter Hawkins | 15 |
| docs/internals/constants.md | George Necula | 11 |
| jax/_src/internal_test_util/export_back_compat_test_data/export_with_specified_sharding.py | George Necula | 11 |
| jax/_src/internal_test_util/export_back_compat_test_data/export_with_unspecified_sharding.py | George Necula | 11 |
| tests/numerics/numerics_test_util_test.py | Peter Hawkins | 10 |
| docs/jax.experimental.sparse.rst | Jake VanderPlas | 8 |
| jax/experimental/key_reuse/__init__.py | Jake VanderPlas | 8 |
| tests/filecheck/jax_mlir_ext.filecheck.py | Peter Hawkins | 8 |
| docs/301/custom-derivatives.md | Matthew Johnson | 8 |
| jaxlib/config_test.py | Peter Hawkins | 7 |
| docs/301/remat.md | Matthew Johnson | 7 |
| jax/_src/lax/lgamma.py | Peter Hawkins | 6 |
| jaxlib/weak_value_interner.cc | Peter Hawkins | 6 |
| jaxlib/py_compile_only_client.h | Peter Hawkins | 6 |
| tests/custom_partitioning_sharding_rule_test.py | Bixia Zheng | 6 |
| jax/experimental/compute_on.py | Yash Katariya | 5 |
| jax/_src/pallas/mosaic/prefetch.py | Robert Dyro | 5 |
| jax/_src/lax/erf.py | Peter Hawkins | 5 |
| jaxlib/weakref_lru_cache_module.cc | Peter Hawkins | 5 |
| jaxlib/weak_value_interner.h | Peter Hawkins | 5 |
| tests/pretty_printer_test.py | Peter Hawkins | 5 |
| jaxlib/reentrant_hash_set.h | Peter Hawkins | 5 |
| docs/301/vjp-objects.md | Matthew Johnson | 5 |
| docs/201/precision.md | Matthew Johnson | 5 |
| docs/301/cookbook.md | Matthew Johnson | 5 |
| docs/whats-new.md | Matthew Johnson | 5 |
| jax/experimental/jax2tf/g3doc/jax_primitives_coverage.md.template | George Necula | 5 |
| docs/jax.experimental.key_reuse.rst | Jake VanderPlas | 4 |
| build_wheel.py | jax authors | 4 |
| jaxlib/strong_lru_cache.cc | Peter Hawkins | 4 |
| tests/numerics/bessel_test.py | Peter Hawkins | 4 |
| jaxlib/reentrant_hash_map.h | Peter Hawkins | 4 |
| jaxlib/nb_class_ptr.h | Peter Hawkins | 4 |
| docs/301/index.rst | Matthew Johnson | 4 |
| docs/501/export.md | Matthew Johnson | 4 |
| docs/501/multiprocess.md | Matthew Johnson | 4 |
| docs/201/index.rst | Matthew Johnson | 4 |
| docs/101/index.rst | Matthew Johnson | 4 |
| examples/ffi/tests/rms_norm_test.py | Dan Foreman-Mackey | 4 |
| jax/_src/tpu/linalg/qdwh.py | Jake VanderPlas | 3 |
| jax/_src/tpu/linalg/stack.py | Jake VanderPlas | 3 |
| docs/jep/25516-effver.md | Jake VanderPlas | 3 |
| docs/jax.typing.rst | Jake VanderPlas | 3 |
| jaxlib/mlir/_mlir_libs/traceback_to_location.cc | Peter Hawkins | 3 |
| jaxlib/weak_value_interner_test.py | Peter Hawkins | 3 |
| jaxlib/reentrant_hash_benchmark.cc | Peter Hawkins | 3 |
| jax/_src/test_warning_util.py | Peter Hawkins | 3 |
| jaxlib/symlink_files.bzl | Peter Hawkins | 3 |
| tests/filecheck/names.filecheck.py | Peter Hawkins | 3 |
| docs/501/compilation-cache.md | Matthew Johnson | 3 |
| docs/501/data-loading.md | Matthew Johnson | 3 |
| docs/601/jax-primitives.md | Matthew Johnson | 3 |
| docs/601/jaxpr.md | Matthew Johnson | 3 |
| docs/101/errors.rst | Matthew Johnson | 3 |
| docs/101/arrays.md | Matthew Johnson | 3 |
| docs/101/convolutions.md | Matthew Johnson | 3 |
| docs/101/default_dtypes.md | Matthew Johnson | 3 |
| docs/101/random.md | Matthew Johnson | 3 |
| docs/101/state.md | Matthew Johnson | 3 |
| docs/101/transformations.md | Matthew Johnson | 3 |
| docs/101/type_promotion.rst | Matthew Johnson | 3 |
| docs/201/aot.md | Matthew Johnson | 3 |
| docs/201/callbacks.md | Matthew Johnson | 3 |
| docs/201/control-flow.md | Matthew Johnson | 3 |
| docs/201/debugging.md | Matthew Johnson | 3 |
| docs/201/jit.md | Matthew Johnson | 3 |
| docs/201/memory-spaces.md | Matthew Johnson | 3 |
| docs/201/placement.md | Matthew Johnson | 3 |
| docs/201/shard-map.md | Matthew Johnson | 3 |
| docs/201/sharding.md | Matthew Johnson | 3 |
| docs/201/slow-compilation.md | Matthew Johnson | 3 |
| docs/301/hijax-types.md | Matthew Johnson | 3 |
| docs/301/refs.md | Matthew Johnson | 3 |
| docs/301/sharding-ad.md | Matthew Johnson | 3 |
| docs/401/ffi.md | Matthew Johnson | 3 |
| docs/401/pallas.md | Matthew Johnson | 3 |
| docs/501/fault-tolerance.rst | Matthew Johnson | 3 |
| docs/501/index.rst | Matthew Johnson | 3 |
| docs/501/shape-polymorphism.md | Matthew Johnson | 3 |
| jax/_src/internal_test_util/export_back_compat_test_data/export_with_multiple_meshes.py | George Necula | 3 |
| third_party/external_deps/BUILD.bazel | Alexandros Theodoridis | 3 |
| third_party/external_deps/external_deps.bzl.tpl | Alexandros Theodoridis | 3 |
| docs/jax.experimental.multihost_utils.rst | Frederic Bastien | 3 |
| jax/experimental/overlap.py | Yash Katariya | 2 |
| jax/tools/evaluate_sampler_bias_test.py | Jake VanderPlas | 2 |
| docs/jax.flatten_util.rst | Jake VanderPlas | 2 |
| docs/_static/type_lattice.svg | Jake VanderPlas | 2 |
| jaxlib/tools/wheel_size_test.py | jax authors | 2 |
| jax/_src/lax/double_word.py | Peter Hawkins | 2 |
| tests/numerics/arithmetic_test.py | Peter Hawkins | 2 |
| third_party/py/rules_python_whl_abi_tags.patch | Peter Hawkins | 2 |
| ci/utilities/setup_portserver.sh | Peter Hawkins | 2 |
| jaxlib/weak_key_weak_value_cache.h | Peter Hawkins | 2 |
| benchmarks/linalg_benchmark.py | Peter Hawkins | 2 |
| jax/_src/hashable_array.py | Peter Hawkins | 2 |
| jaxlib/mlir/_mlir_libs/traceback_to_location.h | Peter Hawkins | 2 |
| jaxlib/custom_call_sharding.h | Peter Hawkins | 2 |
| jaxlib/gpu/gpu_plugin_extension.h | Peter Hawkins | 2 |
| jaxlib/mlir.h | Peter Hawkins | 2 |
| jaxlib/py_client_cpu.h | Peter Hawkins | 2 |
Primary owner = the author with the most commits to a file. Solely owned = files only that author has touched.
| Author | Files owned | Solely owned | Files touched |
|---|---|---|---|
| Peter Hawkins | 516 | 112 | 1,265 |
| Jake VanderPlas | 263 | 23 | 872 |
| Sergei Lebedev | 175 | 6 | 759 |
| Matthew Johnson | 140 | 65 | 495 |
| jax authors | 116 | 12 | 728 |
| Yash Katariya | 77 | 3 | 432 |
| George Necula | 61 | 9 | 386 |
| Adam Paszke | 49 | 13 | 280 |
| Justin Fu | 44 | 23 | 191 |
| Dan Foreman-Mackey | 40 | 11 | 279 |
| Sharad Vikram | 39 | 8 | 264 |
| Roy Frostig | 27 | 3 | 284 |
| 8bitmp3 | 23 | 23 | 48 |
| Skye Wanderman-Milne | 21 | 8 | 148 |
| Nitin Srinivasan | 20 | 4 | 68 |
| Benjamin Chetioui | 19 | 3 | 227 |
| Vladimir Belitskiy | 16 | 3 | 134 |
| Paweł Paruzel | 16 | 1 | 26 |
| Allan Renucci | 14 | 1 | 70 |
| Daniel Suo | 12 | 2 | 241 |
| Ruturaj4 | 12 | 2 | 66 |
| Michael Whittaker | 12 | 10 | 64 |
| Emily Fertig | 11 | 5 | 127 |
| Melissa Weber Mendonça | 11 | 0 | 35 |
| Robert Dyro | 10 | 1 | 127 |
Files touched by the most people — shared code where changes are most likely to need coordination.
| File | Contributors | Primary owner | Owner share |
|---|---|---|---|
| tests/lax_numpy_test.py | 143 | Jake VanderPlas | 31.6% |
| jax/_src/lax/lax.py | 110 | Yash Katariya | 21.2% |
| jax/_src/numpy/lax_numpy.py | 110 | Jake VanderPlas | 48.9% |
| tests/lax_test.py | 85 | Peter Hawkins | 29.0% |
| CHANGELOG.md | 81 | Peter Hawkins | 25.0% |
| tests/api_test.py | 80 | Matthew Johnson | 35.1% |
| tests/BUILD | 80 | Peter Hawkins | 24.0% |
| tests/linalg_test.py | 78 | Peter Hawkins | 39.9% |
| jax/experimental/jax2tf/jax2tf.py | 74 | George Necula | 35.4% |
| jax/_src/api.py | 68 | Yash Katariya | 25.1% |
| jax/_src/interpreters/mlir.py | 65 | Yash Katariya | 25.1% |
| README.md | 64 | Peter Hawkins | 24.2% |
| jax/_src/pallas/mosaic/lowering.py | 62 | Sergei Lebedev | 16.4% |
| jaxlib/BUILD | 61 | Peter Hawkins | 45.6% |
| tests/pjit_test.py | 58 | Yash Katariya | 64.7% |
| jax/_src/config.py | 55 | Peter Hawkins | 17.1% |
| jax/BUILD | 55 | Peter Hawkins | 22.6% |
| jax/_src/lax/linalg.py | 55 | Peter Hawkins | 26.3% |
| tests/pallas/BUILD | 54 | Adam Paszke | 12.1% |
| jax/_src/lax/parallel.py | 53 | Yash Katariya | 23.8% |
How JAX compares with other open source AI and machine learning projects →
How JAX ranks among open source AI and machine learning projects by AI-assisted commits →
Generate this report for your own repository with gitstats-action · All Gallery reports
Publishes the report to GitHub Pages on every push to main: turn on Pages for the gh-pages branch.
# .github/workflows/gitstats.yml
name: GitStats Report
on:
push:
branches: [main]
permissions:
contents: write
jobs:
gitstats:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v7
with:
fetch-depth: 0
- uses: shenxianpeng/gitstats-action@v1
with:
deploy-to-pages: true