Skip to content

Instantly share code, notes, and snippets.

@cgarciae

cgarciae/DLJE_imports.py

Created August 28, 2020 03:25
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save cgarciae/c8b16a2c0fdbac27b4caf5effe61134d to your computer and use it in GitHub Desktop.
Save cgarciae/c8b16a2c0fdbac27b4caf5effe61134d to your computer and use it in GitHub Desktop.
Download ZIP
Imports
Raw
DLJE_imports.py
import numpy as np
import jax
import jax.numpy as jnp
import elegy
import optax
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment