Skip to content

Instantly share code, notes, and snippets.

@diegoferigo
Last active April 2, 2023 15:14
Show Gist options
  • Save diegoferigo/9fd2fdbd9d0de4e39215d2633f8c6a30 to your computer and use it in GitHub Desktop.
Save diegoferigo/9fd2fdbd9d0de4e39215d2633f8c6a30 to your computer and use it in GitHub Desktop.
JAX for scientific programming: an introduction

JAX for scientific programming: an introduction

Binder

name: binder
channels:
- nvidia
- conda-forge
- defaults
dependencies:
- cuda-nvcc
- cudatoolkit
- jax
- jupyter_contrib_nbextensions
- matplotlib
- numpy
- rise
Display the source blob
Display the rendered blob
Raw
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment