To provide 100 JAX exercises over different sections structured as a course or tutorials to teach and learn for beginners, intermediates as well as experts

Related tags

Deep Learningjaxton
Overview

JaxTon

💯 JAX exercises

License GitHub Twitter

Mission 🚀

To provide 100 JAX exercises over different sections structured as a course or tutorials to teach and learn for beginners, intermediates as well as experts.

JAX

The JAX package in Python is a library for high performance and efficient machine learning research.

It is commonly used for various deep learning tasks and runs seamlessly on CPUs, GPUs as well as TPUs.

Exercises 📖

There are a total of 100 JAX exercises divided into 10 sets of Jupyter Notebooks with 10 exercises each. It is recommended to go through the exercises in order but you may start with any set depending on your expertise.

Structured as exercises & tutorials - Choose your style
Suitable for beginners, intermediates & experts - Choose your level
Available on Colab, Kaggle, Binder & GitHub - Choose your platform
Supports running on CPU, GPU & TPU - Choose your backend

Set 01 • JAX Introduction • Beginner • Exercises 1-10

Style Colab Kaggle Binder GitHub
Exercises 1st February, 2022 1st February, 2022 1st February, 2022 1st February, 2022
Solutions 1st February, 2022 1st February, 2022 1st February, 2022 1st February, 2022

Set 02 • Data Operations • Beginner • Exercises 11-20

Style Colab Kaggle Binder GitHub
Exercises 4th February, 2022 4th February, 2022 4th February, 2022 4th February, 2022
Solutions 4th February, 2022 4th February, 2022 4th February, 2022 4th February, 2022

Set 03 • Pseudorandom Numbers • Beginner • Exercises 21-30

Style Colab Kaggle Binder GitHub
Exercises 7th February, 2022 7th February, 2022 7th February, 2022 7th February, 2022
Solutions 7th February, 2022 7th February, 2022 7th February, 2022 7th February, 2022

Set 04 • Just-In-Time (JIT) Compilation • Beginner • Exercises 31-40

Style Colab Kaggle Binder GitHub
Exercises 10th February, 2022 10th February, 2022 10th February, 2022 10th February, 2022
Solutions 10th February, 2022 10th February, 2022 10th February, 2022 10th February, 2022

Set 05 • Control Flows • Beginner • Exercises 41-50

Style Colab Kaggle Binder GitHub
Exercises 13th February, 2022 13th February, 2022 13th February, 2022 13th February, 2022
Solutions 13th February, 2022 13th February, 2022 13th February, 2022 13th February, 2022

Set 06 • Automatic Differentiation • Intermediate • Exercises 51-60

Style Colab Kaggle Binder GitHub
Exercises 16th February, 2022 16th February, 2022 16th February, 2022 16th February, 2022
Solutions 16th February, 2022 16th February, 2022 16th February, 2022 16th February, 2022

Set 07 • Automatic Vectorization • Intermediate • Exercises 61-70

Style Colab Kaggle Binder GitHub
Exercises 19th February, 2022 19th February, 2022 19th February, 2022 19th February, 2022
Solutions 19th February, 2022 19th February, 2022 19th February, 2022 19th February, 2022

Set 08 • Pytrees • Intermediate • Exercises 71-80

Style Colab Kaggle Binder GitHub
Exercises 22nd February, 2022 22nd February, 2022 22nd February, 2022 22nd February, 2022
Solutions 22nd February, 2022 22nd February, 2022 22nd February, 2022 22nd February, 2022

Set 09 • Neural Networks • Expert • Exercises 81-90

Style Colab Kaggle Binder GitHub
Exercises 25th February, 2022 25th February, 2022 25th February, 2022 25th February, 2022
Solutions 25th February, 2022 25th February, 2022 25th February, 2022 25th February, 2022

Set 10 • Capstone Project • Expert • Exercises 91-100

Style Colab Kaggle Binder GitHub
Exercises 28th February, 2022 28th February, 2022 28th February, 2022 28th February, 2022
Solutions 28th February, 2022 28th February, 2022 28th February, 2022 28th February, 2022

The Jupyter Notebooks can also be run locally by cloning the repo and running on your local jupyter server.

git clone https://github.com/vopani/jaxton.git
python3 -m pip install notebook
jupyter notebook

P.S. The notebooks will be periodically updated to improve the exercises and support the latest version.

Contribution 🛠️

Please create an Issue for any improvements, suggestions or errors in the content.

You can also tag @vopani on Twitter for any other queries or feedback.

Credits 🙏

JAX

License 📋

This project is licensed under the Apache License 2.0.

Owner
Rohan Rao
9-time Indian Sudoku Champion | Senior Data Scientist @h2oai | Quadruple Kaggle Grandmaster
Rohan Rao
Code for this paper The Lottery Ticket Hypothesis for Pre-trained BERT Networks.

The Lottery Ticket Hypothesis for Pre-trained BERT Networks Code for this paper The Lottery Ticket Hypothesis for Pre-trained BERT Networks. [NeurIPS

VITA 122 Dec 14, 2022
Code repository for the paper Computer Vision User Entity Behavior Analytics

Computer Vision User Entity Behavior Analytics Code repository for "Computer Vision User Entity Behavior Analytics" Code Description dataset.csv As di

Sameer Khanna 2 Aug 20, 2022
Alphabetical Letter Recognition

DecisionTrees-Image-Classification Alphabetical Letter Recognition In these demo we are using "Decision Trees" Our database is composed by Learning Im

Mohammed Firass 4 Nov 30, 2021
Minimal PyTorch implementation of YOLOv3

A minimal PyTorch implementation of YOLOv3, with support for training, inference and evaluation.

Erik Linder-Norén 6.9k Dec 29, 2022
DeepOBS: A Deep Learning Optimizer Benchmark Suite

DeepOBS - A Deep Learning Optimizer Benchmark Suite DeepOBS is a benchmarking suite that drastically simplifies, automates and improves the evaluation

Aaron Bahde 7 May 12, 2020
A small demonstration of using WebDataset with ImageNet and PyTorch Lightning

A small demonstration of using WebDataset with ImageNet and PyTorch Lightning

Tom 50 Dec 16, 2022
An experimentation and research platform to investigate the interaction of automated agents in an abstract simulated network environments.

CyberBattleSim April 8th, 2021: See the announcement on the Microsoft Security Blog. CyberBattleSim is an experimentation research platform to investi

Microsoft 1.5k Dec 25, 2022
Watch faces morph into each other with StyleGAN 2, StyleGAN, and DCGAN!

FaceMorpher FaceMorpher is an innovative project to get a unique face morph (or interpolation for geeks) on a website. Yes, this means you can see fac

Anish 9 Jun 24, 2022
Pre-training of Graph Augmented Transformers for Medication Recommendation

G-Bert Pre-training of Graph Augmented Transformers for Medication Recommendation Intro G-Bert combined the power of Graph Neural Networks and BERT (B

101 Dec 27, 2022
TensorFlow implementation of PHM (Parameterization of Hypercomplex Multiplication)

Parameterization of Hypercomplex Multiplications (PHM) This repository contains the TensorFlow implementation of PHM (Parameterization of Hypercomplex

Aston Zhang 9 Oct 26, 2022
Development kit for MIT Scene Parsing Benchmark

Development Kit for MIT Scene Parsing Benchmark [NEW!] Our PyTorch implementation is released in the following repository: https://github.com/hangzhao

MIT CSAIL Computer Vision 424 Dec 01, 2022
FIGARO: Generating Symbolic Music with Fine-Grained Artistic Control

FIGARO: Generating Symbolic Music with Fine-Grained Artistic Control by Dimitri von Rütte, Luca Biggio, Yannic Kilcher, Thomas Hofmann FIGARO: Generat

Dimitri 83 Jan 07, 2023
Official pytorch implementation of "Feature Stylization and Domain-aware Contrastive Loss for Domain Generalization" ACMMM 2021 (Oral)

Feature Stylization and Domain-aware Contrastive Loss for Domain Generalization This is an official implementation of "Feature Stylization and Domain-

22 Sep 22, 2022
[ICCV2021] 3DVG-Transformer: Relation Modeling for Visual Grounding on Point Clouds

3DVG-Transformer This repository is for the ICCV 2021 paper "3DVG-Transformer: Relation Modeling for Visual Grounding on Point Clouds" Our method "3DV

22 Dec 11, 2022
OMNIVORE is a single vision model for many different visual modalities

Omnivore: A Single Model for Many Visual Modalities [paper][website] OMNIVORE is a single vision model for many different visual modalities. It learns

Meta Research 451 Dec 27, 2022
An introduction to bioimage analysis - http://bioimagebook.github.io

Introduction to Bioimage Analysis This book tries explain the main ideas of image analysis in a practical and engaging way. It's written primarily for

Bioimage Book 20 Nov 28, 2022
Official implement of Evo-ViT: Slow-Fast Token Evolution for Dynamic Vision Transformer

Evo-ViT: Slow-Fast Token Evolution for Dynamic Vision Transformer This repository contains the PyTorch code for Evo-ViT. This work proposes a slow-fas

YifanXu 53 Dec 05, 2022
Evolutionary Scale Modeling (esm): Pretrained language models for proteins

Evolutionary Scale Modeling This repository contains code and pre-trained weights for Transformer protein language models from Facebook AI Research, i

Meta Research 1.6k Jan 09, 2023
A modern pure-Python library for reading PDF files

pdf A modern pure-Python library for reading PDF files. The goal is to have a modern interface to handle PDF files which is consistent with itself and

6 Apr 06, 2022
Pytorch Implementation of "Diagonal Attention and Style-based GAN for Content-Style disentanglement in image generation and translation" (ICCV 2021)

DiagonalGAN Official Pytorch Implementation of "Diagonal Attention and Style-based GAN for Content-Style Disentanglement in Image Generation and Trans

32 Dec 06, 2022