loeweX / Custom-ConvLayers-PytorchLinks
A reimplementation of 2D Convolutional and Transposed Convolutional Layers in PyTorch, designed for easy modifications and analysis. Includes comprehensive explanations and testing.
☆20Updated last year
Alternatives and similar repositories for Custom-ConvLayers-Pytorch
Users that are interested in Custom-ConvLayers-Pytorch are comparing it to the libraries listed below
Sorting:
- Sparse and discrete interpretability tool for neural networks☆63Updated last year
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆59Updated last year
- Automatic identification of regions in the latent space of a model that correspond to unique concepts, namely to concepts with a semantic…☆14Updated last year
- A lightweight PyTorch implementation of the Transformer-XL architecture proposed by Dai et al. (2019)☆37Updated 2 years ago
- [NeurIPS 2023] Learning Transformer Programs☆162Updated last year
- Functional local implementations of main model parallelism approaches☆96Updated 2 years ago
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- ☆69Updated last year
- Code for papers Linear Algebra with Transformers (TMLR) and What is my Math Transformer Doing? (AI for Maths Workshop, Neurips 2022)☆75Updated last year
- ☆56Updated 10 months ago
- ☆67Updated 6 months ago
- Fastai community entry to 2020 Reproducibility Challenge☆17Updated 2 years ago
- Understanding how features learned by neural networks evolve throughout training☆39Updated 11 months ago
- Jax like function transformation engine but micro, microjax☆33Updated 11 months ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Updated last year
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 3 years ago
- ☆27Updated 2 years ago
- Experiments on GPT-3's ability to fit numerical models in-context.☆14Updated 3 years ago
- Minimum Description Length probing for neural network representations☆20Updated 8 months ago
- ☆34Updated 10 months ago
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Updated last year
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated 2 years ago
- My own attempt at a long context genomics model, leveraging recent advances in long context attention modeling (Flash Attention + other h…☆54Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆150Updated last year
- Embedding Recycling for Language models☆38Updated 2 years ago
- Neural theorem proving tutorial, version II☆39Updated last year
- Differentiable Algorithms and Algorithmic Supervision.☆116Updated 2 years ago