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
A non-linear, non-parametric Machine Learning method capable of modeling complex datasets

Fast Symbolic Regression Symbolic Regression is a non-linear, non-parametric Machine Learning method capable of modeling complex data sets. fastsr aim

VAMSHI CHOWDARY 3 Jun 22, 2022
Membership Inference Attack against Graph Neural Networks

MIA GNN Project Starter If you meet the version mismatch error for Lasagne library, please use following command to upgrade Lasagne library. pip insta

6 Nov 09, 2022
RetinaFace: Deep Face Detection Library in TensorFlow for Python

RetinaFace is a deep learning based cutting-edge facial detector for Python coming with facial landmarks.

Sefik Ilkin Serengil 512 Dec 29, 2022
MacroTools provides a library of tools for working with Julia code and expressions.

MacroTools.jl MacroTools provides a library of tools for working with Julia code and expressions. This includes a powerful template-matching system an

FluxML 278 Dec 11, 2022
一个目标检测的通用框架(不需要cuda编译),支持Yolo全系列(v2~v5)、EfficientDet、RetinaNet、Cascade-RCNN等SOTA网络。

一个目标检测的通用框架(不需要cuda编译),支持Yolo全系列(v2~v5)、EfficientDet、RetinaNet、Cascade-RCNN等SOTA网络。

Haoyu Xu 203 Jan 03, 2023
PyTorch implementation of paper: HPNet: Deep Primitive Segmentation Using Hybrid Representations.

HPNet This repository contains the PyTorch implementation of paper: HPNet: Deep Primitive Segmentation Using Hybrid Representations. Installation The

Siming Yan 42 Dec 07, 2022
Continuous Augmented Positional Embeddings (CAPE) implementation for PyTorch

PyTorch implementation of Continuous Augmented Positional Embeddings (CAPE), by Likhomanenko et al. Enhance your Transformer positional embeddings with easy-to-use augmentations!

Guillermo Cámbara 26 Dec 13, 2022
PyTorch implementation of the WarpedGANSpace: Finding non-linear RBF paths in GAN latent space (ICCV 2021)

Authors official PyTorch implementation of the "WarpedGANSpace: Finding non-linear RBF paths in GAN latent space" [ICCV 2021].

Christos Tzelepis 100 Dec 06, 2022
It is a simple library to speed up CLIP inference up to 3x (K80 GPU)

CLIP-ONNX It is a simple library to speed up CLIP inference up to 3x (K80 GPU) Usage Install clip-onnx module and requirements first. Use this trick !

Gerasimov Maxim 93 Dec 20, 2022
Basit bir burç modülü.

Bu modulu burclar hakkinda gundelik bir sekilde bilgi alin diye yaptim ve sizler icin kullanima sunuyorum. Modulun kullanimi asiri basit: Ornek Kullan

Special 17 Jun 08, 2022
PyTorch implementation of "Learn to Dance with AIST++: Music Conditioned 3D Dance Generation."

Learn to Dance with AIST++: Music Conditioned 3D Dance Generation. Installation pip install -r requirements.txt Prepare Dataset bash data/scripts/pre

Zj Li 8 Sep 07, 2021
blind SQLIpy sebuah alat injeksi sql yang menggunakan waktu sql untuk mendapatkan sebuah server database.

blind SQLIpy Alat blind SQLIpy ini merupakan alat injeksi sql yang menggunakan metode time based blind sql injection metode tersebut membutuhkan waktu

Galih Anggoro Prasetya 4 Feb 24, 2022
a Pytorch easy re-implement of "YOLOX: Exceeding YOLO Series in 2021"

A pytorch easy re-implement of "YOLOX: Exceeding YOLO Series in 2021" 1. Notes This is a pytorch easy re-implement of "YOLOX: Exceeding YOLO Series in

91 Dec 26, 2022
Official PyTorch implementation of "Rapid Neural Architecture Search by Learning to Generate Graphs from Datasets" (ICLR 2021)

Rapid Neural Architecture Search by Learning to Generate Graphs from Datasets This is the official PyTorch implementation for the paper Rapid Neural A

48 Dec 26, 2022
Select, weight and analyze complex sample data

Sample Analytics In large-scale surveys, often complex random mechanisms are used to select samples. Estimates derived from such samples must reflect

samplics 37 Dec 15, 2022
This tool converts a Nondeterministic Finite Automata (NFA) into a Deterministic Finite Automata (DFA)

This tool converts a Nondeterministic Finite Automata (NFA) into a Deterministic Finite Automata (DFA)

Quinn Herden 1 Feb 04, 2022
optimization routines for hyperparameter tuning

Hyperopt: Distributed Hyperparameter Optimization Hyperopt is a Python library for serial and parallel optimization over awkward search spaces, which

Marc Claesen 398 Nov 09, 2022
[Link]mareteutral - pars tradg wth M []

pairs-trading-with-ML Jonathan Larkin, August 2017 One popular strategy classification is Pairs Trading. Though this category of strategies can exhibi

Jonathan Larkin 134 Jan 06, 2023
The official PyTorch code for NeurIPS 2021 ML4AD Paper, "Does Thermal data make the detection systems more reliable?"

MultiModal-Collaborative (MMC) Learning Framework for integrating RGB and Thermal spectral modalities This is the official code for NeurIPS 2021 Machi

NeurAI 12 Nov 02, 2022
Codes for paper "KNAS: Green Neural Architecture Search"

KNAS Codes for paper "KNAS: Green Neural Architecture Search" KNAS is a green (energy-efficient) Neural Architecture Search (NAS) approach. It contain

90 Dec 22, 2022