amithm3 / dvitch Public

PyTorch like API implimentation in JAX

Notifications

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

PyTorch like API implimentation in JAX

No releases published

No packages published