George HotzandGitHub 6e50582e62 working to improve ptx (#3647) ...

* working to improve ptx

* fix compile fail

2024-03-07 12:39:31 -08:00

..

codegen

working to improve ptx (#3647)

2024-03-07 12:39:31 -08:00

features

working to improve ptx (#3647)

2024-03-07 12:39:31 -08:00

nn

Improve type hints for optimizer (#3583)

2024-03-02 07:35:44 -08:00

renderer

working to improve ptx (#3647)

2024-03-07 12:39:31 -08:00

runtime

working to improve ptx (#3647)

2024-03-07 12:39:31 -08:00

shape

Fix symbolic negative floordiv (#3594)

2024-03-03 11:40:52 -05:00

__init__.py

move graph.py and jit.py into features (#3376)

2024-02-12 17:34:34 +01:00

device.py

bring ptx back (#3623)

2024-03-06 13:34:21 -08:00

dtype.py

Rename .sz to .count on DType (#3413)

2024-02-15 15:03:49 +01:00

helpers.py

working to improve ptx (#3647)

2024-03-07 12:39:31 -08:00

lazy.py

lazy.py: cache consts (#3577)

2024-03-02 03:50:05 -08:00

mlops.py

change frontend reduce API to use axis (#3460)

2024-02-21 12:26:17 +01:00

ops.py

fix typing FlopCounter.flops can be sint (#3646)

2024-03-07 12:49:17 -05:00

realize.py

disk runner (prereq for interpreted removal) (#3421)

2024-02-15 18:14:05 +01:00

tensor.py

fix Tensor.to preserves grad.data (#3636)

2024-03-06 21:44:49 -05:00