lebrice
Machine Learning Software Developer at Mila (Quebec Artificial Intelligence Institute)
Mila - Quebec Artificial Intelligence Institute @mila-iqia Montreal, Canada
Pinned Repositories
blurred-GAN
Progressive Growing of GANS using Gaussian Blur
hydra-auto-schema
Adds rich IDE support for Hydra config files
Keras-Progressive-Growing-GANS
(WIP) My Attempt at a clean re-implementation of ["Progressive Growing of GANs for Improved Quality, Stability, and Variation"](https://github.com/tkarras/progressive_growing_of_gans) using the Keras API in TensorFlow. Contributions/Suggestions welcomed!
LLM_api
Testing out SLURM + FastAPI + HuggingFace
MeZO
[NeurIPS 2023] MeZO: Fine-Tuning Language Models with Just Forward Passes. https://arxiv.org/abs/2305.17333
Sequoia
The Research Tree - A playground for research at the intersection of Continual, Reinforcement, and Self-Supervised Learning.
SimpleParsing
Simple, Elegant, Typed Argument Parsing with argparse
torch_jax_interop
Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
Tutorials
Source code for the Mila Tutorials
lebrice's Repositories
lebrice/SimpleParsing
Simple, Elegant, Typed Argument Parsing with argparse
lebrice/torch_jax_interop
Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
lebrice/hydra-auto-schema
Adds rich IDE support for Hydra config files
lebrice/remote-slurm-executor
Plugin for Submitit that allows launching jobs on *remote* SLURM clusters
lebrice/lebrice
lebrice/MeZO
[NeurIPS 2023] MeZO: Fine-Tuning Language Models with Just Forward Passes. https://arxiv.org/abs/2305.17333
lebrice/purejaxql
Simple single-file baselines for Q-Learning in pure-GPU setting
lebrice/spatial
Fooling around, testing out the idea of making a Rust + Python version of gym.spaces. Not a serious project (_yet_ ?)
lebrice/tensor_regression
A small wrapper to use pytest_regressions with Tensors
lebrice/torch-hook-utils
PyTorch Hook utilities
lebrice/causal-conv1d
Causal depthwise conv1d in CUDA, with a PyTorch interface
lebrice/compose
Simple, type-rich function composition in Python (this package contains a single `compose` function)
lebrice/gymnax
RL Environments in JAX 🌍
lebrice/hydra-zen
Create powerful Hydra applications without the yaml files and boilerplate code.
lebrice/hydra_orion_sweeper
Hydra Sweeper Plugin for Orion
lebrice/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
lebrice/mila-docs
Mila technical documentation
lebrice/milatools
Tools to connect to and interact with the Mila cluster
lebrice/mkdocs-autoref-plugin
Small plugin for mkdocs-autoref to allow more IDE-friendly docstrings
lebrice/mktestdocs
Run pytest against markdown files/docstrings.
lebrice/optax
Optax is a gradient processing and optimization library for JAX.
lebrice/orion
Asynchronous Distributed Hyperparameter Optimization.
lebrice/pytest-fixture-tools
Pytest fixture tools
lebrice/ResearchTemplate
WIP: Research Template Repository
lebrice/SARC
FD#11499
lebrice/seqax
seqax = sequence modeling + JAX
lebrice/submitit
Python 3.8+ toolbox for submitting jobs to Slurm
lebrice/titans-pytorch
Unofficial implementation of Titans, SOTA memory for transformers, in Pytorch
lebrice/tool_template
A template repo for a python tool
lebrice/xtils
A collection of utilities for machine learning experiments.