Aakash Kumar Nain's repositories
annotated_research_papers
This repo contains annotated research papers that I found really good and useful
TF_JAX_tutorials
All about the fundamental blocks of TF and JAX!
diffusion_models
All about the fundamentals and working of Diffusion Models
skin_cancer_detection
Detecting cancers on mobile phones
AugMix_TF2
TF2.0 port for Augmix paper
mistral_jax
This is a port of Mistral-7B model in JAX
mistral_keras_jax
Port of Mistral 7B in Keras and JAX
transformers
A collection of resources to study Transformers in depth.
CycleGAN_TF2
Implementation of CycleGAN in tf.keras
benchmark_torch_jax
Models that can be used to benchmark the performance of models built in JAX vs the same models built in PyTorch
keras-core
A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.