| .. |
|
ccl
|
add ccl
|
2022-11-06 15:20:54 +08:00 |
|
dataset
|
polish dataloader
|
2023-01-15 13:30:03 +08:00 |
|
demo
|
optimize concat and split
|
2022-04-04 22:53:20 +08:00 |
|
einops
|
fix pr#355&add unittest
|
2022-11-07 12:01:23 +08:00 |
|
extern
|
LLM support with jtorch
|
2023-03-20 22:33:12 +08:00 |
|
loss3d
|
loss3d: chamfer & emd
|
2021-06-30 15:22:43 +08:00 |
|
math_util
|
add digamma.
|
2023-03-31 11:05:06 +08:00 |
|
models
|
update copyright to 2022
|
2022-04-22 18:01:02 +08:00 |
|
notebook
|
optimize concat and split
|
2022-04-04 22:53:20 +08:00 |
|
other
|
polish log_softmax precision
|
2022-05-18 15:37:09 +08:00 |
|
script
|
update copyright to 2022
|
2022-04-22 18:01:02 +08:00 |
|
src
|
chatglm optimize v2
|
2023-03-31 13:07:47 +08:00 |
|
test
|
Merge branch 'master' of github.com:Jittor/jittor
|
2023-03-31 13:12:08 +08:00 |
|
transform
|
LLM support with jtorch
|
2023-03-20 22:33:12 +08:00 |
|
utils
|
polish cuda thread
|
2023-01-03 17:47:25 +08:00 |
|
__init__.py
|
update load_pytorch.py && add linear
|
2023-03-31 19:45:07 +08:00 |
|
__init__.pyi
|
1. polish index_op doc; 2. polish pyi files
|
2022-09-14 16:09:14 +08:00 |
|
attention.py
|
update copyright to 2022
|
2022-04-22 18:01:02 +08:00 |
|
compile_extern.py
|
support onednn on macOS with intel chips
|
2023-01-05 10:11:35 +08:00 |
|
compiler.py
|
support builtin download of onednn because homebrew no longer provides onednn 2.x
|
2023-01-05 09:38:03 +08:00 |
|
contrib.py
|
LLM support with jtorch
|
2023-03-20 22:33:12 +08:00 |
|
depthwise_conv.py
|
1. fix compilation failure of fp16_compute.h under rocm platform
|
2022-12-17 21:36:54 +08:00 |
|
distributions.py
|
update copyright to 2022
|
2022-04-22 18:01:02 +08:00 |
|
init.py
|
LLM support with jtorch
|
2023-03-20 22:33:12 +08:00 |
|
init_cupy.py
|
update copyright to 2022
|
2022-04-22 18:01:02 +08:00 |
|
linalg.py
|
LLM support with jtorch
|
2023-03-20 22:33:12 +08:00 |
|
lr_scheduler.py
|
update copyright to 2022
|
2022-04-22 18:01:02 +08:00 |
|
misc.py
|
chatglm optimize v2
|
2023-03-31 13:07:47 +08:00 |
|
nn.py
|
update readme && nn.linear
|
2023-03-31 19:45:07 +08:00 |
|
optim.py
|
polish optimzers save&load
|
2023-01-15 13:38:01 +08:00 |
|
pool.py
|
polish backend
|
2022-10-26 14:29:53 +08:00 |
|
pyjt_compiler.py
|
add attr_dict for jittor var
|
2023-03-24 01:53:30 +08:00 |
|
sparse.py
|
update copyright to 2022
|
2022-04-22 18:01:02 +08:00 |
|
version
|
avoid import error when no torch installed
|
2021-06-18 15:56:13 +08:00 |
lzhengning
1119f3cc51