Pytorch Image ModelsPyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
Stars: ✭ 15,232 (+13261.4%)
omdJAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
Stars: ✭ 43 (-62.28%)
uvadlc notebooksRepository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2022/Spring 2022
Stars: ✭ 901 (+690.35%)
koclipKoCLIP: Korean port of OpenAI CLIP, in Flax
Stars: ✭ 80 (-29.82%)
TPU-MobilenetSSDEdge TPU Accelerator / Multi-TPU + MobileNet-SSD v2 + Python + Async + LattePandaAlpha/RaspberryPi3/LaptopPC
Stars: ✭ 82 (-28.07%)
PyprobmlPython code for "Machine learning: a probabilistic perspective" (2nd edition)
Stars: ✭ 4,197 (+3581.58%)
TensorMONKA collection of deep learning models (PyTorch implemtation)
Stars: ✭ 21 (-81.58%)
score flowOfficial code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)
Stars: ✭ 49 (-57.02%)
jax-resnetImplementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
Stars: ✭ 61 (-46.49%)
get-started-with-JAXThe 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 of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
Stars: ✭ 229 (+100.88%)
awesome-computer-vision-modelsA list of popular deep learning models related to classification, segmentation and detection problems
Stars: ✭ 419 (+267.54%)
jax-modelsUnofficial JAX implementations of deep learning research papers
Stars: ✭ 108 (-5.26%)
Transformers🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Stars: ✭ 55,742 (+48796.49%)
MixNet-PyTorchConcise, Modular, Human-friendly PyTorch implementation of MixNet with Pre-trained Weights.
Stars: ✭ 16 (-85.96%)
jax-rlJAX implementations of core Deep RL algorithms
Stars: ✭ 61 (-46.49%)
TensorrtxImplementation of popular deep learning networks with TensorRT network definition API
Stars: ✭ 3,456 (+2931.58%)
tiny-tpuSmall-scale Tensor Processing Unit built on an FPGA
Stars: ✭ 61 (-46.49%)
graphsignalGraphsignal Python agent
Stars: ✭ 158 (+38.6%)
efficientdetPyTorch Implementation of the state-of-the-art model for object detection EfficientDet [pre-trained weights provided]
Stars: ✭ 21 (-81.58%)
rA9JAX-based Spiking Neural Network framework
Stars: ✭ 60 (-47.37%)
quarkdetQuarkDet lightweight object detection in PyTorch .Real-Time Object Detection on Mobile Devices.
Stars: ✭ 82 (-28.07%)
FlaxengineFlax Engine – multi-platform 3D game engine
Stars: ✭ 3,127 (+2642.98%)
mlp-gpt-jaxA GPT, made only of MLPs, in Jax
Stars: ✭ 53 (-53.51%)
mobilevit-pytorchA PyTorch implementation of "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer".
Stars: ✭ 349 (+206.14%)
flaxOptimizersA collection of optimizers, some arcane others well known, for Flax.
Stars: ✭ 21 (-81.58%)
SAN[ECCV 2020] Scale Adaptive Network: Learning to Learn Parameterized Classification Networks for Scalable Input Images
Stars: ✭ 41 (-64.04%)
FlaxSamplesCollection of example projects for Flax Engine
Stars: ✭ 50 (-56.14%)
MobilePose-PiMobilePose deployment for Raspberry Pi
Stars: ✭ 15 (-86.84%)
MobilePoseLight-weight Single Person Pose Estimator
Stars: ✭ 588 (+415.79%)
image embeddingsUsing efficientnet to provide embeddings for retrieval
Stars: ✭ 107 (-6.14%)
bigbrother-specsResearch and specification for Big Brother protocol
Stars: ✭ 13 (-88.6%)
Tensor2tensorLibrary of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
Stars: ✭ 11,865 (+10307.89%)
GPJaxA didactic Gaussian process package for researchers in Jax.
Stars: ✭ 159 (+39.47%)
GSOCRepository for Google Summer of Code 2019 https://summerofcode.withgoogle.com/projects/#4662790671826944
Stars: ✭ 61 (-46.49%)
food-detection-yolov5🍔🍟🍗 Food analysis baseline with Theseus. Integrate object detection, image classification and multi-class semantic segmentation. 🍞🍖🍕
Stars: ✭ 68 (-40.35%)
hftaBoost hardware utilization for ML training workloads via Inter-model Horizontal Fusion
Stars: ✭ 18 (-84.21%)
ml-testing-acceleratorsTesting framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
Stars: ✭ 45 (-60.53%)
EinopsDeep learning operations reinvented (for pytorch, tensorflow, jax and others)
Stars: ✭ 4,022 (+3428.07%)
AdanetFast and flexible AutoML with learning guarantees.
Stars: ✭ 3,340 (+2829.82%)
jax-cfdComputational Fluid Dynamics in JAX
Stars: ✭ 399 (+250%)
JaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Stars: ✭ 15,579 (+13565.79%)
berserkerBerserker - BERt chineSE woRd toKenizER
Stars: ✭ 17 (-85.09%)
EfficientUNetPlusPlusDecoder architecture based on the UNet++. Combining residual bottlenecks with depthwise convolutions and attention mechanisms, it outperforms the UNet++ in a coronary artery segmentation task, while being significantly more computationally efficient.
Stars: ✭ 37 (-67.54%)
Thinc🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
Stars: ✭ 2,422 (+2024.56%)
FoolboxA Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
Stars: ✭ 2,108 (+1749.12%)
ADAMADAM implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.
Stars: ✭ 51 (-55.26%)
ShinRLShinRL: A Library for Evaluating RL Algorithms from Theoretical and Practical Perspectives (Deep RL Workshop 2021)
Stars: ✭ 30 (-73.68%)
googlecodelabsTPU ile Yapay Sinir Ağlarınızı Çok Daha Hızlı Eğitin
Stars: ✭ 116 (+1.75%)