Commit Graph

1614 Commits

Author SHA1 Message Date
lzhengning b824ab83c0 update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
Haoyang Peng 4825cead21
add Gamma Distribution on Cuda (#435)
* add digamma.

* add tdir.

* add gamma distribution.

* add test for gamma distribution.

* update location.

* add api directly into jt.__init__
2023-04-25 14:25:10 +08:00
lzhengning 6bf14ee650 pip publish for ADAN optimizer, related PR https://github.com/Jittor/jittor/pull/430 2023-04-10 14:03:04 +08:00
xingyuxie ab2e2e11d4 add adan optimizer 2023-04-10 13:59:57 +08:00
zhouwy19 9c4924881d
Update AWESOME-JITTOR-LIST.md 2023-04-07 11:27:35 +08:00
Dun Liang e395ae7791 update version with weight norm 2023-04-05 02:16:23 +08:00
Haoyang Peng 2e67d0eb8f
add support for weight_norm and histc. (#420)
* add support for weight_norm and histc.

* add docs.

* fix typo.

---------

Co-authored-by: Exusial <Exusial>
2023-04-05 21:59:20 +08:00
Dun Liang d11d2c816e polish include 2023-04-05 02:12:17 +08:00
GanymedeNil 2b79a2c496
Update swap.cc (#429)
fix getpid not found
2023-04-05 21:46:21 +08:00
Dun Liang 0464661920 update version 2023-04-04 23:49:10 +08:00
Dun Liang 45f617cb54 add pid location for swap 2023-04-04 23:15:42 +08:00
Dun Liang 5cf9c10c92 fix windows virtual env bug 2023-04-04 19:30:22 +08:00
Dun Liang 3bcdfc97a9 update version 2023-04-04 17:37:24 +08:00
Dun Liang 57d5b52b18 Merge branch 'master' of github.com:Jittor/jittor 2023-04-04 17:37:07 +08:00
lzhengning 10be18fbc5 support python3.11 on mac 2023-04-04 22:59:36 +08:00
Dun Liang 0036671536 add mem opt doc 2023-04-04 17:36:56 +08:00
Dun Liang 33b091e041 support python3.11 2023-04-04 14:54:40 +08:00
Dun Liang b88ffdbdf5 fix windows load pth bug 2023-04-04 13:12:32 +08:00
Dun Liang 2ea015061c polish swap iter 2023-04-03 12:55:01 +08:00
Dun Liang 7a83dcf09e fix windows mkl bug 2023-04-03 07:56:44 +08:00
Dun Liang 16b7966a9a set default amp level 2023-04-03 01:27:55 +08:00
Dun Liang df8628a3a5 Merge branch 'master' of github.com:Jittor/jittor 2023-04-02 07:18:24 +08:00
Haoyang Peng 779274653f
Gamma (#427)
* add digamma.

* add tdir.
2023-04-02 13:57:09 +08:00
Dun Liang a19936f954 new feature: save_mem 2023-04-02 07:17:55 +08:00
Dun Liang 00b7783c46 Merge branch 'master' of github.com:Jittor/jittor 2023-03-31 21:14:20 +08:00
Dun Liang e969103399 polish to_float interface 2023-03-31 21:13:59 +08:00
Dun Liang 3300a3e4f3 polish fp16 amp_level3 2023-03-31 21:07:43 +08:00
lzhengning 463fdcc46d rollback version 2023-03-31 19:45:07 +08:00
lzhengning 1119f3cc51 update readme && nn.linear 2023-03-31 19:45:07 +08:00
lzhengning 8a9f0bb904 update load_pytorch.py && add linear 2023-03-31 19:45:07 +08:00
Dun Liang b451a1eb85 Merge branch 'master' of github.com:Jittor/jittor 2023-03-31 13:12:08 +08:00
Dun Liang faa5386d82 chatglm optimize v2 2023-03-31 13:07:47 +08:00
Exusial f79f2d37e8 add digamma. 2023-03-31 11:05:06 +08:00
Dun Liang 1e00883ed1 add attr_dict for jittor var 2023-03-24 01:53:30 +08:00
Dun Liang 5638de9eca LLM support and add version 2023-03-22 19:51:11 +08:00
Dun Liang 07f8bd4aff Merge branch 'master' of github.com:Jittor/jittor 2023-03-22 19:47:26 +08:00
Dun Liang db8679748c fp32_guard && ternary broadcast 2023-03-22 19:41:36 +08:00
lzhengning 50c859f60c support build triu/tril of an inf matrix
The original implementation results in nan because zeros * inf = nan.
Suggest to use jt.ternary op instead of multiply a mask matrix.
2023-03-22 19:19:36 +08:00
lzhengning 576b0c9e03 feat: jt.outer & jt.nn.silu 2023-03-21 15:56:21 +08:00
Dun Liang 90850d36ad LLM support with jtorch 2023-03-20 22:33:12 +08:00
lzhengning 2f0e243e01 feat: support dim=None in Var.squeeze to squeeze all dimensions of size 1 2023-03-07 11:33:16 +08:00
LiDongyang 3703fe4421 polish optimzers save&load 2023-01-15 13:38:01 +08:00
Exusial daa0862896 fix multiple sequence load. 2023-01-15 13:37:07 +08:00
Brian Pugh 5dba84ee67 Allow for nn.Sequential to be sliced. 2023-01-15 13:36:29 +08:00
Exusial d1eeec5f89 Add nan fp16 support. 2023-01-15 13:35:10 +08:00
Dun Liang b25e62f1bb polish dataloader 2023-01-15 13:30:03 +08:00
Dun Liang 6fbffb7c62 jtorch compatible 2023-01-06 20:48:04 +08:00
Dun Liang 437a720500 add multinomial operator 2023-01-06 19:39:58 +08:00
Dun Liang 5a4ae74a3a add index_select 2023-01-06 17:08:30 +08:00
lzhengning 2b24d436e3 update version 2023-01-05 11:05:35 +08:00