FJFormer icon indicating copy to clipboard operation
FJFormer copied to clipboard

verified publisher icon erfanzar
→

Metadata

paralleled/unparalleled computational with FJFormer

Results 2 FJFormer issues
Sort by recently updated
recently updated
newest added

[fix] Apply `with_sharding_constraint` recursively to pytrees

[generated by copilot] This pull request refactors and enhances the handling of sharding constraints in the `eformer/escale/partition/constraints.py` file. The most notable changes include renaming and improving an existing function for...

dvruette avatar dvruette

JAX 0.7.0+ Incompatibility

Using EasyDel on JAX >=0.7.0 causes an error ``` RuntimeError: Failed to import easydel.modules.auto because of the following error (look up to see its traceback): module 'jax._src.pjit' has no attribute...

Ueaj-Kerman avatar Ueaj-Kerman

← Metadata

24
Stars
4
Forks
Watchers

Owner

verified publisher icon erfanzar

Metadata

paralleled/unparalleled computational with FJFormer

Back