Activity
Fix flaky vmap test tolerance.
Fix flaky vmap test tolerance.
Force push
Fix flaky vmap test tolerance.
Fix flaky vmap test tolerance.
Force push
Fix flaky vmap test tolerance.
Fix flaky vmap test tolerance.
Force push
Fix flaky vmap test tolerance.
Fix flaky vmap test tolerance.
Force push
[nnx] add more flaxlib types
[nnx] add more flaxlib types
Force push
[nnx] refactor GraphDef
[nnx] refactor GraphDef
Force push
[nnx] remove transforms.deprecated
[nnx] remove transforms.deprecated
Force push
[nnx] refactor UpdateContext
[nnx] refactor UpdateContext
[nnx] add more flaxlib types
[nnx] add more flaxlib types
Force push
[nnx] add more flaxlib types
[nnx] add more flaxlib types
Force push
[nnx] add more flaxlib types
[nnx] add more flaxlib types
Force push
[nnx] refactor GraphDef
[nnx] refactor GraphDef
Force push
[nnx] add more flaxlib types
[nnx] add more flaxlib types
Force push
[nnx] refactor GraphDef
[nnx] refactor GraphDef
Force push
[nnx] refactor GraphDef
[nnx] refactor GraphDef
Force push
[nnx] remove transforms.deprecated
[nnx] remove transforms.deprecated
Force push
[nnx] refactor GraphDef
[nnx] refactor GraphDef
Force push
[nnx] remove transforms.deprecated
[nnx] remove transforms.deprecated
Force push
[nnx] support Array leaves in graph nodes
[nnx] support Array leaves in graph nodes
Force push
[nnx] refactor GraphDef
[nnx] refactor GraphDef
Force push