CoffeeBeforeArch / pytest
Simple pytest examples
☆12Updated 2 years ago
Related projects: ⓘ
- ML/DL Math and Method notes☆56Updated 9 months ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated last year
- Reasons, motivations, and resources that help pursuing the doctoral degree in Computer Science☆12Updated 2 years ago
- Helper scripts I use to run many experiments in the morning to check at night☆19Updated 3 years ago
- ☆16Updated 3 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 2 years ago
- Introductory lecture on Pytorch☆15Updated 2 years ago
- Experiment of using Tangent to autodiff triton☆66Updated 7 months ago
- ☆17Updated 4 months ago
- This is a port of Mistral-7B model in JAX☆29Updated 2 months ago
- Custom kernels in Triton language for accelerating LLMs☆14Updated 5 months ago
- ☆30Updated 3 months ago
- ☆56Updated 2 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆43Updated 2 years ago
- ☆35Updated 2 years ago
- notebooks of cool EBM visualizations☆16Updated 3 years ago
- Karras et al. (2022) diffusion models for PyTorch☆13Updated last year
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆57Updated last month
- Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to …☆45Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆41Updated 3 years ago
- Learning about CUDA by writing PTX code.☆28Updated 6 months ago
- Slides and recordings of talks hosted by our community☆18Updated 2 months ago
- Sharpened Cosine Distance implementation in PyTorch☆10Updated 2 years ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆34Updated 2 months ago
- A simple Transformer where the softmax has been replaced with normalization☆17Updated 4 years ago
- Learn CUDA with PyTorch☆11Updated last month
- Hacks for PyTorch☆17Updated last year
- ☆15Updated 3 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- Lightning HPO & Training Studio App☆17Updated last year