haliax 1.4.dev365__tar.gz → 1.4.dev366__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (112) hide show
  1. {haliax-1.4.dev365 → haliax-1.4.dev366}/PKG-INFO +2 -1
  2. {haliax-1.4.dev365 → haliax-1.4.dev366}/pyproject.toml +2 -1
  3. haliax-1.4.dev366/src/haliax/__about__.py +1 -0
  4. haliax-1.4.dev365/src/haliax/__about__.py +0 -1
  5. {haliax-1.4.dev365 → haliax-1.4.dev366}/.coveragerc +0 -0
  6. {haliax-1.4.dev365 → haliax-1.4.dev366}/.flake8 +0 -0
  7. {haliax-1.4.dev365 → haliax-1.4.dev366}/.github/workflows/publish_dev.yaml +0 -0
  8. {haliax-1.4.dev365 → haliax-1.4.dev366}/.github/workflows/run_pre_commit.yaml +0 -0
  9. {haliax-1.4.dev365 → haliax-1.4.dev366}/.github/workflows/run_quick_levanter_tests.yaml +0 -0
  10. {haliax-1.4.dev365 → haliax-1.4.dev366}/.github/workflows/run_tests.yaml +0 -0
  11. {haliax-1.4.dev365 → haliax-1.4.dev366}/.gitignore +0 -0
  12. {haliax-1.4.dev365 → haliax-1.4.dev366}/.pre-commit-config.yaml +0 -0
  13. {haliax-1.4.dev365 → haliax-1.4.dev366}/.readthedocs.yaml +0 -0
  14. {haliax-1.4.dev365 → haliax-1.4.dev366}/AGENTS.md +0 -0
  15. {haliax-1.4.dev365 → haliax-1.4.dev366}/CONTRIBUTING.md +0 -0
  16. {haliax-1.4.dev365 → haliax-1.4.dev366}/LICENSE +0 -0
  17. {haliax-1.4.dev365 → haliax-1.4.dev366}/README.md +0 -0
  18. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/api.md +0 -0
  19. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/broadcasting.md +0 -0
  20. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/cheatsheet.md +0 -0
  21. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/css/material.css +0 -0
  22. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/css/mkdocstrings.css +0 -0
  23. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/faq.md +0 -0
  24. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/figures/data_parallel_mesh.png +0 -0
  25. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/figures/data_parallel_mesh_replicated.png +0 -0
  26. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/figures/device_mesh_1d.png +0 -0
  27. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/figures/device_mesh_1d_zero.png +0 -0
  28. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/figures/device_mesh_2d.png +0 -0
  29. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/figures/device_mesh_2d_batch_partitioned.png +0 -0
  30. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/figures/device_mesh_2d_data_replicated.png +0 -0
  31. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/figures/device_mesh_2d_data_replicated_mlp_partitioned.png +0 -0
  32. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/figures/device_mesh_2d_intermediate_fully_partitioned.png +0 -0
  33. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/figures/device_mesh_2d_zero.png +0 -0
  34. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/fp8.md +0 -0
  35. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/index.md +0 -0
  36. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/indexing.md +0 -0
  37. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/matmul.md +0 -0
  38. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/nn.md +0 -0
  39. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/partitioning.md +0 -0
  40. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/rearrange.ipynb +0 -0
  41. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/rearrange.md +0 -0
  42. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/requirements.txt +0 -0
  43. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/scan.md +0 -0
  44. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/state-dict.md +0 -0
  45. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/tutorial.md +0 -0
  46. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/typing.md +0 -0
  47. {haliax-1.4.dev365 → haliax-1.4.dev366}/docs/vmap.md +0 -0
  48. {haliax-1.4.dev365 → haliax-1.4.dev366}/mkdocs.yml +0 -0
  49. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/__init__.py +0 -0
  50. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/_src/__init__.py +0 -0
  51. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/_src/compile_utils.py +0 -0
  52. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/_src/dot.py +0 -0
  53. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/_src/einsum.py +0 -0
  54. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/_src/fp8.py +0 -0
  55. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/_src/parsing.py +0 -0
  56. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/_src/rearrange.py +0 -0
  57. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/_src/scan.py +0 -0
  58. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/_src/state_dict.py +0 -0
  59. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/_src/util.py +0 -0
  60. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/axis.py +0 -0
  61. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/core.py +0 -0
  62. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/debug.py +0 -0
  63. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/haxtyping.py +0 -0
  64. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/hof.py +0 -0
  65. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/jax_utils.py +0 -0
  66. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/__init__.py +0 -0
  67. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/activations.py +0 -0
  68. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/attention.py +0 -0
  69. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/conv.py +0 -0
  70. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/dropout.py +0 -0
  71. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/embedding.py +0 -0
  72. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/linear.py +0 -0
  73. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/loss.py +0 -0
  74. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/mlp.py +0 -0
  75. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/normalization.py +0 -0
  76. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/pool.py +0 -0
  77. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/nn/scan.py +0 -0
  78. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/ops.py +0 -0
  79. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/partitioning.py +0 -0
  80. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/quantization.py +0 -0
  81. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/random.py +0 -0
  82. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/specialized_fns.py +0 -0
  83. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/state_dict.py +0 -0
  84. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/tree_util.py +0 -0
  85. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/types.py +0 -0
  86. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/util.py +0 -0
  87. {haliax-1.4.dev365 → haliax-1.4.dev366}/src/haliax/wrap.py +0 -0
  88. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/core_test.py +0 -0
  89. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_attention.py +0 -0
  90. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_axis.py +0 -0
  91. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_conv.py +0 -0
  92. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_debug.py +0 -0
  93. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_dot.py +0 -0
  94. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_dtype_typing.py +0 -0
  95. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_einsum.py +0 -0
  96. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_fp8.py +0 -0
  97. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_hof.py +0 -0
  98. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_int8.py +0 -0
  99. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_namedarray_typing.py +0 -0
  100. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_nn.py +0 -0
  101. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_ops.py +0 -0
  102. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_parsing.py +0 -0
  103. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_partitioning.py +0 -0
  104. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_pool.py +0 -0
  105. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_random.py +0 -0
  106. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_rearrange.py +0 -0
  107. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_scan.py +0 -0
  108. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_scatter_gather.py +0 -0
  109. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_specialized_fns.py +0 -0
  110. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_state_dict.py +0 -0
  111. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_tree_util.py +0 -0
  112. {haliax-1.4.dev365 → haliax-1.4.dev366}/tests/test_utils.py +0 -0
{haliax-1.4.dev365 → haliax-1.4.dev366}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: haliax
3
- Version: 1.4.dev365
3
+ Version: 1.4.dev366
4
4
  Summary: Named Tensors for Legible Deep Learning in JAX
5
5
  Project-URL: Homepage, https://github.com/stanford-crfm/haliax
6
6
  Project-URL: Bug Tracker, https://github.com/stanford-crfm/haliax/issues/
@@ -29,6 +29,7 @@ Requires-Dist: mkdocs>=1.4.3; extra == 'dev'
29
29
  Requires-Dist: mkdocstrings-python>=1.1.2; extra == 'dev'
30
30
  Requires-Dist: mkdocstrings>=0.22.0; extra == 'dev'
31
31
  Requires-Dist: mypy>=0.910; extra == 'dev'
32
+ Requires-Dist: pre-commit; extra == 'dev'
32
33
  Requires-Dist: pygments; extra == 'dev'
33
34
  Requires-Dist: pymdown-extensions; extra == 'dev'
34
35
  Requires-Dist: pytest>=7.4.0; extra == 'dev'
{haliax-1.4.dev365 → haliax-1.4.dev366}/pyproject.toml RENAMED
@@ -38,7 +38,8 @@ dev=["pytest >= 7.4.0", "mypy >= 0.910", "mkdocs >= 1.4.3", "mkdocs-material >=
38
38
  "pymdown-extensions",
39
39
  "pygments",
40
40
  "pymdown-extensions",
41
- "chex>=0.1.86"
41
+ "chex>=0.1.86",
42
+ "pre-commit",
42
43
  ]
43
44
 
44
45
 
haliax-1.4.dev366/src/haliax/__about__.py ADDED
@@ -0,0 +1 @@
1
+ __version__ = "1.4.dev366"
haliax-1.4.dev365/src/haliax/__about__.py DELETED
@@ -1 +0,0 @@
1
- __version__ = "1.4.dev365"
{haliax-1.4.dev365 → haliax-1.4.dev366}/.coveragerc RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/.flake8 RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/.gitignore RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/AGENTS.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/CONTRIBUTING.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/LICENSE RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/README.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/docs/api.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/docs/faq.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/docs/fp8.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/docs/index.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/docs/matmul.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/docs/nn.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/docs/scan.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/docs/typing.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/docs/vmap.md RENAMED
File without changes
{haliax-1.4.dev365 → haliax-1.4.dev366}/mkdocs.yml RENAMED
File without changes