Repository navigation
Commit ebab535
committed
Major context rework and dryml.{ml_framework} addition
* Total rework of context management
* Add new dryml.tf, dryml.torch, and dryml.jax modules which can add
special methods for those frameworks1 parent b2e0889 commit ebab535
24 files changed
Lines changed: 1853 additions & 130 deletions
File tree
- src/dryml
- context
- jax
- plain
- tf
- torch
- jax
- tf
- torch
- tests
- jax
- tf
- torch
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | 1 | | |
2 | | - | |
| 2 | + | |
| 3 | + | |
3 | 4 | | |
4 | 5 | | |
5 | 6 | | |
| |||
13 | 14 | | |
14 | 15 | | |
15 | 16 | | |
| 17 | + | |
16 | 18 | | |
17 | 19 | | |
18 | 20 | | |
| |||
0 commit comments