Giters

amithm3 / dvitch

PyTorch like API implimentation in JAX

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

dvitch

PyTorch like API implimentation in JAX

About

PyTorch like API implimentation in JAX

License:Apache License 2.0

Languages

Language:Jupyter Notebook 93.1%Language:Python 6.9%