haliax 1.4.dev405__tar.gz → 1.4.dev406__tar.gz

haliax-1.4.dev406/.agents/projects/api_parity.md

@@ -0,0 +1,215 @@
+# JAX NumPy API Parity
+This document tracks JAX NumPy functions not yet wrapped by Haliax.
+
+APIs that don't translate well to named tensors are intentionally omitted here. This includes dtype constructors, raw array converters (e.g. `from_dlpack`), indexing helpers like `c_`/`r_`, and functions whose JAX counterparts already work with `NamedArray` out of the box. Basic array construction is handled by `haliax.named`, so functions such as `array` or `asarray` are not listed.
+
+## numpy
+- [ ] `allclose`
+- [ ] `amin`
+- [ ] `append`
+- [ ] `apply_along_axis`
+- [ ] `apply_over_axes`
+- [ ] `argpartition`
+- [ ] `argwhere`
+- [ ] `array_equal`
+- [ ] `array_equiv`
+- [ ] `array_split`
+- [ ] `astype`
+- [ ] `atan2`
+- [ ] `average`
+- [ ] `bartlett`
+- [ ] `bitwise_count`
+- [ ] `bitwise_invert`
+- [ ] `bitwise_left_shift`
+- [ ] `bitwise_right_shift`
+- [ ] `blackman`
+- [ ] `block`
+- [ ] `broadcast_shapes`
+- [ ] `can_cast`
+- [ ] `choose`
+- [ ] `column_stack`
+- [ ] `compress`
+- [ ] `concat`
+- [ ] `convolve`
+- [ ] `copysign`
+- [ ] `corrcoef`
+- [ ] `correlate`
+- [ ] `count_nonzero`
+- [ ] `cov`
+- [ ] `cumulative_prod`
+- [ ] `cumulative_sum`
+- [ ] `delete`
+- [ ] `diag`
+- [ ] `diag_indices`
+- [ ] `diag_indices_from`
+- [ ] `diagflat`
+- [ ] `diagonal`
+- [ ] `diff`
+- [ ] `digitize`
+- [ ] `dsplit`
+- [ ] `dstack`
+- [ ] `ediff1d`
+- [ ] `einsum_path`
+- [ ] `empty`
+- [ ] `empty_like`
+- [ ] `expand_dims`
+- [ ] `extract`
+- [ ] `eye`
+- [ ] `fill_diagonal`
+- [ ] `flatnonzero`
+- [ ] `flip`
+- [ ] `fliplr`
+- [ ] `flipud`
+- [ ] `gcd`
+- [ ] `gradient`
+- [ ] `hamming`
+- [ ] `hanning`
+- [ ] `heaviside`
+- [ ] `histogram`
+- [ ] `histogram2d`
+- [ ] `histogram_bin_edges`
+- [ ] `histogramdd`
+- [ ] `hsplit`
+- [ ] `hstack`
+- [ ] `identity`
+- [ ] `indices`
+- [ ] `inner`
+- [ ] `insert`
+- [ ] `interp`
+- [ ] `intersect1d`
+- [ ] `iscomplexobj`
+- [ ] `isin`
+- [ ] `iterable`
+- [ ] `ix_`
+- [ ] `kaiser`
+- [ ] `kron`
+- [ ] `lcm`
+- [ ] `ldexp`
+- [ ] `lexsort`
+- [ ] `mask_indices`
+- [ ] `median`
+- [ ] `meshgrid`
+- [ ] `mgrid`
+- [ ] `modf`
+- [ ] `moveaxis`
+- [ ] `nan_to_num`
+- [ ] `nanargmax`
+- [ ] `nanargmin`
+- [ ] `nancumprod`
+- [ ] `nancumsum`
+- [ ] `nanmax`
+- [ ] `nanmean`
+- [ ] `nanmedian`
+- [ ] `nanmin`
+- [ ] `nanpercentile`
+- [ ] `nanprod`
+- [ ] `nanquantile`
+- [ ] `nanstd`
+- [ ] `nansum`
+- [ ] `nanvar`
+- [ ] `nonzero`
+- [ ] `ogrid`
+- [ ] `packbits`
+- [ ] `partition`
+- [ ] `percentile`
+- [ ] `permute_dims`
+- [ ] `piecewise`
+- [ ] `place`
+- [ ] `poly`
+- [ ] `polyadd`
+- [ ] `polyder`
+- [ ] `polydiv`
+- [ ] `polyfit`
+- [ ] `polyint`
+- [ ] `polymul`
+- [ ] `polysub`
+- [ ] `polyval`
+- [ ] `pow`
+- [ ] `promote_types`
+- [ ] `put`
+- [ ] `put_along_axis`
+- [ ] `quantile`
+- [ ] `ravel_multi_index`
+- [ ] `reshape`
+- [ ] `resize`
+- [ ] `result_type`
+- [ ] `rollaxis`
+- [ ] `roots`
+- [ ] `rot90`
+- [ ] `select`
+- [ ] `setdiff1d`
+- [ ] `setxor1d`
+- [ ] `spacing`
+- [ ] `squeeze`
+- [ ] `swapaxes`
+- [ ] `take_along_axis`
+- [ ] `transpose`
+- [ ] `trapezoid`
+- [ ] `tri`
+- [ ] `tril_indices`
+- [ ] `tril_indices_from`
+- [ ] `trim_zeros`
+- [ ] `triu_indices`
+- [ ] `triu_indices_from`
+- [ ] `union1d`
+- [ ] `unpackbits`
+- [ ] `unravel_index`
+- [ ] `unstack`
+- [ ] `unwrap`
+- [ ] `vander`
+- [ ] `vsplit`
+- [ ] `vstack`
+
+## fft
+- [ ] `fft`
+- [ ] `fft2`
+- [ ] `fftfreq`
+- [ ] `fftn`
+- [ ] `fftshift`
+- [ ] `hfft`
+- [ ] `ifft`
+- [ ] `ifft2`
+- [ ] `ifftn`
+- [ ] `ifftshift`
+- [ ] `ihfft`
+- [ ] `irfft`
+- [ ] `irfft2`
+- [ ] `irfftn`
+- [ ] `rfft`
+- [ ] `rfft2`
+- [ ] `rfftfreq`
+- [ ] `rfftn`
+
+## linalg
+- [ ] `cholesky`
+- [ ] `cond`
+- [ ] `cross`
+- [ ] `det`
+- [ ] `diagonal`
+- [ ] `eig`
+- [ ] `eigh`
+- [ ] `eigvals`
+- [ ] `eigvalsh`
+- [ ] `inv`
+- [ ] `lstsq`
+- [ ] `matmul`
+- [ ] `matrix_norm`
+- [ ] `matrix_power`
+- [ ] `matrix_rank`
+- [ ] `matrix_transpose`
+- [ ] `norm`
+- [ ] `outer`
+- [ ] `pinv`
+- [ ] `qr`
+- [ ] `slogdet`
+- [ ] `solve`
+- [ ] `svd`
+- [ ] `svdvals`
+- [ ] `tensordot`
+- [ ] `tensorinv`
+- [ ] `tensorsolve`
+- [ ] `trace`
+- [ ] `vdot`
+- [ ] `vecdot`
+- [ ] `vecmat`
+- [ ] `vector_norm`

{haliax-1.4.dev405 → haliax-1.4.dev406}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haliax
-Version: 1.4.dev405
+Version: 1.4.dev406
 Summary: Named Tensors for Legible Deep Learning in JAX
 Project-URL: Homepage, https://github.com/stanford-crfm/haliax
 Project-URL: Bug Tracker, https://github.com/stanford-crfm/haliax/issues/

haliax-1.4.dev406/src/haliax/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.4.dev406"

haliax-1.4.dev405/src/haliax/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.4.dev405"