AakashKumarNain / TF_JAX_tutorials
All about the fundamental blocks of TF and JAX!
☆271Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for TF_JAX_tutorials
- 100 exercises to learn JAX☆569Updated 2 years ago
- Host repository for the "Reproducible Deep Learning" PhD course☆405Updated 2 years ago
- All about the fundamentals and working of Diffusion Models☆152Updated last year
- Lightning Bits: Engineering for Researchers repo☆130Updated 2 years ago
- The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series o…☆661Updated 11 months ago
- A curated list of awesome fastai projects/blog posts/tutorials/etc.☆164Updated 2 years ago
- ML Research paper summaries, annotated papers and implementation walkthroughs☆113Updated 2 years ago
- A walkthrough of transformer architecture code☆318Updated 9 months ago
- Notebooks for "Probabilistic Machine Learning" book☆202Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.☆389Updated last week
- A library to inspect and extract intermediate layers of PyTorch models.☆470Updated 2 years ago
- For optimization algorithm research and development.☆449Updated this week
- deep learning with pytorch lightning☆0Updated 2 weeks ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆175Updated 2 years ago
- ☆137Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated last year
- MinT: Minimal Transformer Library and Tutorials☆248Updated 2 years ago
- Notebooks for the Practicals at the Deep Learning Indaba 2022.☆175Updated 7 months ago
- Lightweight Hyperparameter Optimization 🚂☆145Updated 2 months ago
- GitHub Repo with various ML/AI/DS resources that I find useful☆455Updated 4 months ago
- Kaggle Pipeline for tabular data competitions☆204Updated 4 months ago
- This repository contains an overview of important follow-up works based on the original Vision Transformer (ViT) by Google.☆149Updated 2 years ago
- A High Level API for Deep Learning in JAX☆470Updated last year
- ☆196Updated this week
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆408Updated 7 months ago
- Highly commented implementations of Transformers in PyTorch☆128Updated last year
- My repo for training neural nets using pytorch-lightning and hydra☆214Updated 3 months ago
- CS 7301: Spring 2021 Course on Advanced Topics in Optimization in Machine Learning☆174Updated 3 years ago
- ☆391Updated last month