Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 2, 2024

• Improved static arg test

  young-geng

  committed
  e0d0d4a

Commits on Oct 24, 2024

• Merge pull request #5 from Sea-Snell/main

  Show description for fb8c5d1
  young-geng

  authored
  fb8c5d1

• Use native API for physical axis splitting

  Sea-Snell

  authored
  8cf9cea

Commits on Oct 16, 2024

• Bug fix

  young-geng

  committed
  2525c01

Commits on Aug 13, 2024

• Improved global mesh handling in eval_shape

  young-geng

  committed
  33db96f

Commits on Jun 26, 2024

• Bump version for release

  young-geng

  committed
  992570b

• Improved access to raw jitted function.

  young-geng

  committed
  75905d4

• Added utils for easy access of jitted function

  young-geng

  committed
  f86cfba

Commits on Mar 16, 2024

• Improve static argument doc and test

  young-geng

  committed
  530fdbe

Commits on Mar 7, 2024

• Bump version for release

  young-geng

  committed
  3467665

Commits on Mar 6, 2024

• Removed unused attention module

  young-geng

  committed
  8906022

• fixed typo in documentation

  Sea-Snell

  authored
  f8f31ce

• Update readme

  young-geng

  committed
  5ba0dbd

• Docs and exmaples

  young-geng

  committed
  b20f094

Commits on Mar 5, 2024

• Update examples

  young-geng

  committed
  67c9c46

• More documentations

  young-geng

  committed
  f26e833

Commits on Mar 1, 2024

• Change sharding tests to use sharding constraint and annotation functions instead of class methods

  young-geng

  committed
  1103dc9

Commits on Feb 29, 2024

• Improved documentation and updated version

  young-geng

  committed
  ba56de7

Commits on Feb 28, 2024

• More tests and changed sjit arg name

  young-geng

  committed
  a98fe30

Commits on Feb 27, 2024

• Bumped version for release.

  young-geng

  committed
  c67f60a

• Bug fix and more tests

  young-geng

  committed
  d80e8b1

• Added tests

  young-geng

  committed
  18a93df

Commits on Feb 26, 2024

• Merge pull request #4 from young-geng/sharding_annotation

  Show description for 0d9d5f3
  Sea-Snell

  authored
  0d9d5f3

• Implement sharding annotation options

  young-geng

  committed
  eb0e8d9

Commits on Feb 11, 2024

• Rename sharded_jit to sjit in readme

  young-geng

  committed
  1da5cfb

• Added util to print pytree structure and sharding rule based on callable policy.

  young-geng

  committed
  b7e0cba

Commits on Feb 6, 2024

• Bump version for pypi release

  young-geng

  committed
  42eb597

• Local data to global array bug fix

  young-geng

  committed
  87d99b5

• Default module imports

  young-geng

  committed
  06b1718

• Added data sharding utils

  young-geng

  committed
  c20fcc5

Commits on Feb 3, 2024

• Update version and some examples

  young-geng

  committed
  3824206

Commits on Jan 7, 2024

• Merge pull request #2 from Sea-Snell/main

  Show description for f650e8a
  young-geng

  authored
  f650e8a

• Merge branch 'young-geng:main' into main

  Sea-Snell

  authored
  1f24c86

• fixed bug with gather functions

  Sea-Snell

  committed
  2f3df6d

Commits on Nov 12, 2023

• Merge pull request #1 from Sea-Snell/main

  Show description for 5f278d4
  young-geng

  authored
  5f278d4

Pagination

PreviousNext