amithm3 / dvitch Public

PyTorch like API implimentation in JAX

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
.idea		.idea
src		src
.gitignore		.gitignore
LICENCE		LICENCE
README.md		README.md
main.py		main.py
test.ipynb		test.ipynb
test.py		test.py
test2.py		test2.py

Repository files navigation

dvitch

PyTorch like API implimentation in JAX