Giters

cvoelcker / jax_modules

Collection of useful jax snippets and modules to build more complex differentiable code

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

jax_modules

Collection of useful jax snippets and modules to build more complex differentiable code

About

Collection of useful jax snippets and modules to build more complex differentiable code

License:MIT License