poets-ai / elegy

A High Level API for Deep Learning in JAX

Home Page:https://poets-ai.github.io/elegy/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

poets-ai/elegy Issues