Skip to content
Snippets Groups Projects
Select Git revision
  • 60-remove-shardedarray
  • EiffL/joss-paper
  • JZ_branch
  • fix_fft
  • fix_halo_again
  • fix_slice_unpad
  • implement-jax-versions
  • joss-paper
  • main default
  • non-contiguous-transpose-axis
  • one-transpositions-for-slabs
  • permute-transpose
  • push-to-pypi
  • u/ASKabalan/clean-up
  • u/ASKabalan/demo_branch
  • u/ASKabalan/transpose_ops
  • update-jax-array-api
17 results
You can move around the graph by using the arrow keys.
Created with Raphaël 2.2.010Feb21Jan20181421Dec1698629Nov2424Oct232221201917127630Sep20Aug1624Jul19189878754328Jun2718131131May64230Apr292827282725821Mar2115Jan29Jan25Nov2423212019161211fix60-remove-shard…60-remove-shardedarrayrun ruffupdate READMESrun ruff formatterupdate exposed APIupdate APIRemove jaxdecomp.config.transpose_axis_contiguous_2add error message for JAX sharding bugmajor cleanupupdate testsupdate cudecomp implsremove sharded arrayupdate jax primitivesMerge pull request #57 from DifferentiableUniverseInitiative/ASKabalan_patchbump version to 0.2.4Merge pull request #56 from DifferentiableUniverseInitiative/ASKabalan_patchupdate from tree_map to tree.mapQuick fix .. ShardedArray with None sharding returns ShardedArrayMerge remote-tracking branch 'origin/main' into ASKabalan_patchMerge pull request #55 from DifferentiableUniverseInitiative/54-updating-cudecomp-submoduleupdate cuDecomp versionadd halo grad testsome fixesMerge pull request #53 from DifferentiableUniverseInitiative/52-implement-shardmapped-opsadd typing extensionformatupdate formatter to ruffupdate testsadd tree map for fft utilsadd tree map for pfft promote to complexupdate transposeupdate Haloupdate FFTReplace custom_partitionning class with cleaner decorater .. add ShardedArrayMerge pull request #49 from DifferentiableUniverseInitiative/ASKabalan_patchbumping versionMerge remote-tracking branch 'origin/main' into ASKabalan_patchMerge pull request #51 from DifferentiableUniverseInitiative/quick-fixesbump jax version in dependenciesFix gradient issue with PFFT3D