Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
src		src
.gitignore		.gitignore
README.md		README.md
environment.yml		environment.yml

Repository files navigation

JAX playground

Playground for learning jax in python

JAX

JAX is a library for high-performance numerical computing in python with autograd and XLA support.

Setup

Install miniconda
Create conda environment using conda env create -f environment.yml
Activate conda environment using conda activate jax-playground

About

Playground for learning jax in python

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%