Cristian Garcia's repositories

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Language:PythonLicense:Apache-2.0Stargazers:0Issues:0Issues:0

ciclo

A functional training loops library for JAX

Language:PythonLicense:MITStargazers:81Issues:0Issues:0
Language:ShellStargazers:0Issues:0Issues:0

nanoGPT-jax

The simplest, fastest repository for training/finetuning medium-sized GPTs.

Language:Jupyter NotebookLicense:MITStargazers:22Issues:0Issues:0

flax

Flax is a neural network library for JAX that is designed for flexibility.

Language:PythonLicense:Apache-2.0Stargazers:1Issues:0Issues:0

nnx

Neural Networks for JAX

Language:PythonLicense:MITStargazers:74Issues:0Issues:0
Language:RustStargazers:1Issues:0Issues:0

maxtext-nnx

A simple, performant and scalable Jax LLM!

Language:PythonLicense:Apache-2.0Stargazers:2Issues:0Issues:0
Language:PythonStargazers:0Issues:0Issues:0
Language:JinjaStargazers:0Issues:0Issues:0
Language:PythonStargazers:3Issues:0Issues:0

simple-pytree

A dead simple Python package for creating custom JAX pytree objects

Language:PythonLicense:MITStargazers:59Issues:0Issues:0
Language:CSSStargazers:0Issues:0Issues:0
Language:Jupyter NotebookLicense:MITStargazers:24Issues:0Issues:0

pypeln

Concurrent data pipelines in Python >>>

Language:PythonLicense:MITStargazers:1520Issues:0Issues:0
Stargazers:0Issues:0Issues:0

stable-diffusion-webui

Stable Diffusion web UI

License:AGPL-3.0Stargazers:0Issues:0Issues:0
Language:PythonLicense:MITStargazers:4Issues:0Issues:0

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Language:PythonLicense:Apache-2.0Stargazers:0Issues:0Issues:0

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

License:Apache-2.0Stargazers:0Issues:0Issues:0

community-events

Place where folks can contribute to 🤗 community events

Stargazers:0Issues:0Issues:0

jax_metrics

A metrics library for the JAX ecosystem

Language:PythonLicense:MITStargazers:35Issues:0Issues:0
Stargazers:0Issues:0Issues:0
Language:RustStargazers:0Issues:0Issues:0
License:MITStargazers:0Issues:0Issues:0

treeo

A small library for creating and manipulating custom JAX Pytree classes

Language:PythonLicense:MITStargazers:59Issues:0Issues:0

treex

A Pytree Module system for Deep Learning in JAX

Language:PythonLicense:MITStargazers:215Issues:0Issues:0

nbstripout

strip output from Jupyter and IPython notebooks

License:NOASSERTIONStargazers:0Issues:0Issues:0

minGPT-jax

A minimal JAX re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Language:PythonLicense:MITStargazers:3Issues:0Issues:0
Language:Jupyter NotebookLicense:MITStargazers:5Issues:0Issues:0