Pinned Repositories
ciclo
A functional training loops library for JAX
jax_metrics
A metrics library for the JAX ecosystem
karma
An MVC framework for Unity3D
phi
Functional Programming + Python - Pain
pypeln
Concurrent data pipelines in Python >>>
simple-pytree
A dead simple Python package for creating custom JAX pytree objects
tensorbuilder
TensorBuilder is a TensorFlow library enables you to easily create complex deep neural networks by leveraging the phi DSL to help define their structure.
treex
A Pytree Module system for Deep Learning in JAX
flax
Flax is a neural network library for JAX that is designed for flexibility.
elegy
A High Level API for Deep Learning in JAX
cgarciae's Repositories
cgarciae/treex
A Pytree Module system for Deep Learning in JAX
cgarciae/einop
cgarciae/dataget
A framework-agnostic datasets library for Machine Learning research and education.
cgarciae/resume
Machine Learning Engineer with background in Maths & Physics
cgarciae/simple-detr
cgarciae/jax-streams
cgarciae/minGPT-jax
A minimal JAX re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
cgarciae/simple-ddpm
cgarciae/flax-tools
cgarciae/simple-attention
cgarciae/diffusion_models
A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch
cgarciae/flax-xor-demo
cgarciae/awesome-jax
JAX - A curated list of resources https://github.com/google/jax
cgarciae/clearml-demo
cgarciae/codespaces-test
cgarciae/CommonLoopUtils
CLU lets you write beautiful training loops in JAX.
cgarciae/dataset_iterator
cgarciae/ddpm-jax
cgarciae/einops
Deep learning operations reinvented (for pytorch, tensorflow, chainer, gluon and others)
cgarciae/equinox
Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.
cgarciae/gitignore
A collection of useful .gitignore templates
cgarciae/markdown-cv
a simple template to write your CV in a readable markdown file and use CSS to publish/print it.
cgarciae/nbstripout
strip output from Jupyter and IPython notebooks
cgarciae/pac
cgarciae/quansight-labs-site
Development site and blog for Quansight Labs
cgarciae/scikit-learn
scikit-learn: machine learning in Python
cgarciae/stable-diffusion-jax
cgarciae/t5x
cgarciae/TF_JAX_tutorials
All about the fundamental blocks of TF and JAX!
cgarciae/tree-math
Mathematical operations for JAX pytrees