Lucidrains 系列项目源码解析（目录）
【代码】Lucidrains 系列项目源码解析（目录）
·
.\lucidrains\Adan-pytorch\adan_pytorch\adan.py.\lucidrains\Adan-pytorch\adan_pytorch\__init__.py.\lucidrains\Adan-pytorch\setup.py.\lucidrains\adjacent-attention-network\adjacent_attention_network\adjacent_attention_network.py.\lucidrains\adjacent-attention-network\adjacent_attention_network\__init__.py- Adjacent Attention Network
.\lucidrains\adjacent-attention-network\setup.py.\lucidrains\agent-attention-pytorch\agent_attention_pytorch\agent_attention_pytorch.py.\lucidrains\agent-attention-pytorch\agent_attention_pytorch\agent_transformer.py.\lucidrains\agent-attention-pytorch\agent_attention_pytorch\__init__.py.\lucidrains\agent-attention-pytorch\setup.py.\lucidrains\all-normalization-transformer\all_normalization_transformer\all_normalization_transformer.py.\lucidrains\all-normalization-transformer\all_normalization_transformer\autoregressive_wrapper.py.\lucidrains\all-normalization-transformer\all_normalization_transformer\__init__.py- Data source
- Transformer with Normalized Attention
.\lucidrains\all-normalization-transformer\train_enwik8.py.\lucidrains\alphafold2\alphafold2_pytorch\alphafold2.py.\lucidrains\alphafold2\alphafold2_pytorch\constants.py.\lucidrains\alphafold2\alphafold2_pytorch\embeds.py.\lucidrains\alphafold2\alphafold2_pytorch\mlm.py.\lucidrains\alphafold2\alphafold2_pytorch\reversible.py.\lucidrains\alphafold2\alphafold2_pytorch\rotary.py.\lucidrains\alphafold2\alphafold2_pytorch\utils.py.\lucidrains\alphafold2\alphafold2_pytorch\__init__.py.\lucidrains\alphafold2\scripts\refinement.py.\lucidrains\alphafold2\setup.py.\lucidrains\alphafold2\tests\test_attention.py.\lucidrains\alphafold2\tests\test_utils.py.\lucidrains\alphafold2\training_scripts\datasets\trrosetta.py.\lucidrains\alphafold2\training_scripts\datasets\__init__.py.\lucidrains\alphafold2\training_scripts\deepspeed.py.\lucidrains\alphafold2\training_scripts\lightning.py.\lucidrains\alphafold2\train_end2end.py.\lucidrains\alphafold2\train_pre.py.\lucidrains\AMIE-pytorch\AMIE_pytorch\AMIE_pytorch.py.\lucidrains\AMIE-pytorch\AMIE_pytorch\__init__.py.\lucidrains\AMIE-pytorch\setup.py.\lucidrains\anymal-belief-state-encoder-decoder-pytorch\anymal_belief_state_encoder_decoder_pytorch\networks.py.\lucidrains\anymal-belief-state-encoder-decoder-pytorch\anymal_belief_state_encoder_decoder_pytorch\ppo.py.\lucidrains\anymal-belief-state-encoder-decoder-pytorch\anymal_belief_state_encoder_decoder_pytorch\running.py.\lucidrains\anymal-belief-state-encoder-decoder-pytorch\anymal_belief_state_encoder_decoder_pytorch\trainer.py.\lucidrains\anymal-belief-state-encoder-decoder-pytorch\anymal_belief_state_encoder_decoder_pytorch\__init__.py.\lucidrains\anymal-belief-state-encoder-decoder-pytorch\setup.py.\lucidrains\AoA-pytorch\aoa_pytorch\aoa_pytorch.py.\lucidrains\AoA-pytorch\aoa_pytorch\__init__.py.\lucidrains\AoA-pytorch\setup.py.\lucidrains\attention-tensorflow-mesh\attention_tensorflow_mesh\attention_tensorflow_mesh.py.\lucidrains\attention-tensorflow-mesh\attention_tensorflow_mesh\__init__.py- Attention for Tensorflow Mesh
.\lucidrains\attention-tensorflow-mesh\setup.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\attend.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\audiolm_pytorch.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\data.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\encodec.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\hubert_kmeans.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\optimizer.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\soundstream.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\t5.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\trainer.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\utils.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\version.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\vq_wav2vec.py.\lucidrains\audiolm-pytorch\audiolm_pytorch\__init__.py.\lucidrains\audiolm-pytorch\setup.py.\lucidrains\autoregressive-linear-attention-cuda\autoregressive_linear_attention_cuda\autoregressive_linear_attention_cuda.py.\lucidrains\autoregressive-linear-attention-cuda\autoregressive_linear_attention_cuda\__init__.py- Linear Attention - Autoregressive CUDA kernel (wip)
.\lucidrains\autoregressive-linear-attention-cuda\setup.py.\lucidrains\axial-attention\axial_attention\axial_attention.py.\lucidrains\axial-attention\axial_attention\reversible.py.\lucidrains\axial-attention\axial_attention\__init__.py- Axial Attention
.\lucidrains\axial-attention\setup.py.\lucidrains\axial-positional-embedding\axial_positional_embedding\axial_positional_embedding.py.\lucidrains\axial-positional-embedding\axial_positional_embedding\__init__.py- Axial Positional Embedding
.\lucidrains\axial-positional-embedding\setup.py.\lucidrains\bidirectional-cross-attention\bidirectional_cross_attention\bidirectional_cross_attention.py.\lucidrains\bidirectional-cross-attention\bidirectional_cross_attention\__init__.py- Bidirectional Cross Attention
.\lucidrains\bidirectional-cross-attention\setup.py.\lucidrains\big-sleep\big_sleep\biggan.py.\lucidrains\big-sleep\big_sleep\big_sleep.py.\lucidrains\big-sleep\big_sleep\cli.py.\lucidrains\big-sleep\big_sleep\ema.py.\lucidrains\big-sleep\big_sleep\resample.py.\lucidrains\big-sleep\big_sleep\version.py.\lucidrains\big-sleep\big_sleep\__init__.py.\lucidrains\big-sleep\setup.py.\lucidrains\big-sleep\test\multi_prompt_minmax.py.\lucidrains\bit-diffusion\bit_diffusion\bit_diffusion.py.\lucidrains\bit-diffusion\bit_diffusion\__init__.py.\lucidrains\bit-diffusion\setup.py.\lucidrains\block-recurrent-transformer-pytorch\block_recurrent_transformer_pytorch\block_recurrent_transformer_pytorch.py.\lucidrains\block-recurrent-transformer-pytorch\block_recurrent_transformer_pytorch\__init__.py- Data source
.\lucidrains\block-recurrent-transformer-pytorch\setup.py.\lucidrains\block-recurrent-transformer-pytorch\train.py.\lucidrains\bottleneck-transformer-pytorch\bottleneck_transformer_pytorch\bottleneck_transformer_pytorch.py.\lucidrains\bottleneck-transformer-pytorch\bottleneck_transformer_pytorch\__init__.py.\lucidrains\bottleneck-transformer-pytorch\setup.py.\lucidrains\BS-RoFormer\bs_roformer\attend.py.\lucidrains\BS-RoFormer\bs_roformer\bs_roformer.py.\lucidrains\BS-RoFormer\bs_roformer\mel_band_roformer.py.\lucidrains\BS-RoFormer\bs_roformer\__init__.py.\lucidrains\BS-RoFormer\setup.py.\lucidrains\byol-pytorch\byol_pytorch\byol_pytorch.py.\lucidrains\byol-pytorch\byol_pytorch\trainer.py.\lucidrains\byol-pytorch\byol_pytorch\__init__.py- Pytorch-lightning example script
.\lucidrains\byol-pytorch\examples\lightning\train.py.\lucidrains\byol-pytorch\setup.py.\lucidrains\CALM-pytorch\CALM_pytorch\CALM.py.\lucidrains\CALM-pytorch\CALM_pytorch\sampling_utils.py.\lucidrains\CALM-pytorch\CALM_pytorch\__init__.py.\lucidrains\CALM-pytorch\setup.py.\lucidrains\charformer-pytorch\charformer_pytorch\charformer_pytorch.py.\lucidrains\charformer-pytorch\charformer_pytorch\__init__.py.\lucidrains\charformer-pytorch\setup.py.\lucidrains\chroma-pytorch\chroma_pytorch\chroma_pytorch.py.\lucidrains\chroma-pytorch\chroma_pytorch\semantic_conditioner.py.\lucidrains\chroma-pytorch\chroma_pytorch\__init__.py.\lucidrains\chroma-pytorch\setup.py.\lucidrains\classifier-free-guidance-pytorch\classifier_free_guidance_pytorch\attend.py.\lucidrains\classifier-free-guidance-pytorch\classifier_free_guidance_pytorch\bge.py.\lucidrains\classifier-free-guidance-pytorch\classifier_free_guidance_pytorch\classifier_free_guidance_pytorch.py.\lucidrains\classifier-free-guidance-pytorch\classifier_free_guidance_pytorch\open_clip.py.\lucidrains\classifier-free-guidance-pytorch\classifier_free_guidance_pytorch\t5.py.\lucidrains\classifier-free-guidance-pytorch\classifier_free_guidance_pytorch\__init__.py- Classifier Free Guidance - Pytorch
.\lucidrains\classifier-free-guidance-pytorch\setup.py.\lucidrains\CoCa-pytorch\coca_pytorch\coca_pytorch.py.\lucidrains\CoCa-pytorch\coca_pytorch\__init__.py.\lucidrains\CoCa-pytorch\setup.py.\lucidrains\coco-lm-pytorch\coco_lm_pytorch\coco_lm_pytorch.py.\lucidrains\coco-lm-pytorch\coco_lm_pytorch\__init__.py.\lucidrains\coco-lm-pytorch\setup.py- coffee-neural-network
.\lucidrains\CoLT5-attention\colt5_attention\attend.py.\lucidrains\CoLT5-attention\colt5_attention\coor_descent.py.\lucidrains\CoLT5-attention\colt5_attention\topk.py.\lucidrains\CoLT5-attention\colt5_attention\transformer_block.py.\lucidrains\CoLT5-attention\colt5_attention\triton_coor_descent.py.\lucidrains\CoLT5-attention\colt5_attention\vit.py.\lucidrains\CoLT5-attention\colt5_attention\__init__.py.\lucidrains\CoLT5-attention\setup.py.\lucidrains\complex-valued-transformer\complex_valued_transformer\attend.py.\lucidrains\complex-valued-transformer\complex_valued_transformer\autoregressive_wrapper.py.\lucidrains\complex-valued-transformer\complex_valued_transformer\complex_valued_transformer.py.\lucidrains\complex-valued-transformer\complex_valued_transformer\__init__.py- Data source
.\lucidrains\complex-valued-transformer\setup.py.\lucidrains\complex-valued-transformer\train.py.\lucidrains\compositional-attention-pytorch\compositional_attention_pytorch\compositional_attention_pytorch.py.\lucidrains\compositional-attention-pytorch\compositional_attention_pytorch\__init__.py.\lucidrains\compositional-attention-pytorch\setup.py.\lucidrains\compressive-transformer-pytorch\compressive_transformer_pytorch\autoregressive_wrapper.py.\lucidrains\compressive-transformer-pytorch\compressive_transformer_pytorch\compressive_transformer_pytorch.py.\lucidrains\compressive-transformer-pytorch\compressive_transformer_pytorch\__init__.py- Data source
.\lucidrains\compressive-transformer-pytorch\examples\enwik8_simple\train.py.\lucidrains\compressive-transformer-pytorch\setup.py.\lucidrains\conformer\conformer\conformer.py.\lucidrains\conformer\conformer\__init__.py.\lucidrains\conformer\setup.py.\lucidrains\contrastive-learner\contrastive_learner\contrastive_learner.py.\lucidrains\contrastive-learner\contrastive_learner\__init__.py.\lucidrains\contrastive-learner\setup.py.\lucidrains\coordinate-descent-attention\coordinate_descent_attention\autoregressive_wrapper.py.\lucidrains\coordinate-descent-attention\coordinate_descent_attention\coordinate_descent_attention.py.\lucidrains\coordinate-descent-attention\coordinate_descent_attention\__init__.py- Data source
- Coordinate Descent Attention (wip)
.\lucidrains\coordinate-descent-attention\setup.py.\lucidrains\coordinate-descent-attention\train.py.\lucidrains\cross-transformers-pytorch\cross_transformers_pytorch\cross_transformers_pytorch.py.\lucidrains\cross-transformers-pytorch\cross_transformers_pytorch\__init__.py.\lucidrains\cross-transformers-pytorch\setup.py.\lucidrains\DALLE-pytorch\dalle_pytorch\attention.py.\lucidrains\DALLE-pytorch\dalle_pytorch\dalle_pytorch.py.\lucidrains\DALLE-pytorch\dalle_pytorch\distributed_backends\deepspeed_backend.py.\lucidrains\DALLE-pytorch\dalle_pytorch\distributed_backends\distributed_backend.py.\lucidrains\DALLE-pytorch\dalle_pytorch\distributed_backends\dummy_backend.py.\lucidrains\DALLE-pytorch\dalle_pytorch\distributed_backends\horovod_backend.py.\lucidrains\DALLE-pytorch\dalle_pytorch\distributed_backends\__init__.py.\lucidrains\DALLE-pytorch\dalle_pytorch\distributed_utils.py.\lucidrains\DALLE-pytorch\dalle_pytorch\loader.py.\lucidrains\DALLE-pytorch\dalle_pytorch\reversible.py.\lucidrains\DALLE-pytorch\dalle_pytorch\transformer.py.\lucidrains\DALLE-pytorch\dalle_pytorch\vae.py.\lucidrains\DALLE-pytorch\dalle_pytorch\version.py.\lucidrains\DALLE-pytorch\dalle_pytorch\__init__.py.\lucidrains\DALLE-pytorch\generate.py- DALL-E in Pytorch
.\lucidrains\DALLE-pytorch\setup.py.\lucidrains\DALLE-pytorch\train_dalle.py.\lucidrains\DALLE-pytorch\train_vae.py- DALLE2 Training Configurations
.\lucidrains\DALLE2-pytorch\dalle2_pytorch\cli.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\dalle2_pytorch.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\dataloaders\decoder_loader.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\dataloaders\prior_loader.py- Dataloaders
.\lucidrains\DALLE2-pytorch\dalle2_pytorch\dataloaders\simple_image_only_dataloader.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\dataloaders\__init__.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\optimizer.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\trackers.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\trainer.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\train_configs.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\utils.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\version.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\vqgan_vae.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\vqgan_vae_trainer.py.\lucidrains\DALLE2-pytorch\dalle2_pytorch\__init__.py- Diffusion Prior
.\lucidrains\DALLE2-pytorch\setup.py.\lucidrains\DALLE2-pytorch\train_decoder.py.\lucidrains\DALLE2-pytorch\train_diffusion_prior.py.\lucidrains\ddpm-ipa-protein-generation\ddpm_ipa_protein_generation\ddpm_ipa_protein_generation.py.\lucidrains\ddpm-ipa-protein-generation\ddpm_ipa_protein_generation\__init__.py- DDPM + IPA for Protein Structure and Sequence Generation (wip)
.\lucidrains\ddpm-ipa-protein-generation\setup.py.\lucidrains\ddpm-proteins\cache.py.\lucidrains\ddpm-proteins\ddpm_proteins\ddpm_proteins.py.\lucidrains\ddpm-proteins\ddpm_proteins\utils.py.\lucidrains\ddpm-proteins\ddpm_proteins\__init__.py- Denoising Diffusion Probabilistic Model for Proteins
.\lucidrains\ddpm-proteins\setup.py.\lucidrains\ddpm-proteins\train.py.\lucidrains\deep-daze\deep_daze\cli.py.\lucidrains\deep-daze\deep_daze\deep_daze.py.\lucidrains\deep-daze\deep_daze\version.py.\lucidrains\deep-daze\deep_daze\__init__.py- Deep Daze
.\lucidrains\deep-daze\setup.py.\lucidrains\deep-linear-network\deep_linear_network\deep_linear_network.py.\lucidrains\deep-linear-network\deep_linear_network\__init__.py.\lucidrains\deep-linear-network\setup.py.\lucidrains\deformable-attention\deformable_attention\deformable_attention_1d.py.\lucidrains\deformable-attention\deformable_attention\deformable_attention_2d.py.\lucidrains\deformable-attention\deformable_attention\deformable_attention_3d.py.\lucidrains\deformable-attention\deformable_attention\__init__.py.\lucidrains\deformable-attention\setup.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\attend.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\classifier_free_guidance.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\continuous_time_gaussian_diffusion.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\denoising_diffusion_pytorch.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\denoising_diffusion_pytorch_1d.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\elucidated_diffusion.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\fid_evaluation.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\guided_diffusion.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\karras_unet.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\karras_unet_1d.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\karras_unet_3d.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\learned_gaussian_diffusion.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\simple_diffusion.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\version.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\v_param_continuous_time_gaussian_diffusion.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\weighted_objective_gaussian_diffusion.py.\lucidrains\denoising-diffusion-pytorch\denoising_diffusion_pytorch\__init__.py.\lucidrains\denoising-diffusion-pytorch\setup.py- Differentiable Signed Distance Function Rendering - Pytorch (wip)
.\lucidrains\diffusion-policy\diffusion_policy\diffusion_policy.py.\lucidrains\diffusion-policy\diffusion_policy\__init__.py.\lucidrains\diffusion-policy\setup.py.\lucidrains\discrete-key-value-bottleneck-pytorch\discrete_key_value_bottleneck_pytorch\discrete_key_value_bottleneck.py.\lucidrains\discrete-key-value-bottleneck-pytorch\discrete_key_value_bottleneck_pytorch\__init__.py.\lucidrains\discrete-key-value-bottleneck-pytorch\setup.py- Distilling Knowledge from Reader to Retriever
- Dreamcraft3d - Pytorch (wip)
.\lucidrains\egnn-pytorch\denoise_sparse.py.\lucidrains\egnn-pytorch\egnn_pytorch\egnn_pytorch.py.\lucidrains\egnn-pytorch\egnn_pytorch\egnn_pytorch_geometric.py.\lucidrains\egnn-pytorch\egnn_pytorch\utils.py.\lucidrains\egnn-pytorch\egnn_pytorch\__init__.py.\lucidrains\egnn-pytorch\setup.py.\lucidrains\egnn-pytorch\tests\test_equivariance.py.\lucidrains\einops-exts\einops_exts\einops_exts.py.\lucidrains\einops-exts\einops_exts\torch.py.\lucidrains\einops-exts\einops_exts\__init__.py- Einops Extensions
.\lucidrains\einops-exts\setup.py.\lucidrains\electra-pytorch\electra_pytorch\electra_pytorch.py.\lucidrains\electra-pytorch\electra_pytorch\__init__.py.\lucidrains\electra-pytorch\examples\glue\download.py.\lucidrains\electra-pytorch\examples\glue\metrics.py.\lucidrains\electra-pytorch\examples\glue\processors.py.\lucidrains\electra-pytorch\examples\glue\run.py.\lucidrains\electra-pytorch\examples\glue\utils.py.\lucidrains\electra-pytorch\pretraining\openwebtext\arg.py.\lucidrains\electra-pytorch\pretraining\openwebtext\dataset.py.\lucidrains\electra-pytorch\pretraining\openwebtext\preprocess.py.\lucidrains\electra-pytorch\pretraining\openwebtext\pretrain.py.\lucidrains\electra-pytorch\pretraining\openwebtext\tokenization.py.\lucidrains\electra-pytorch\setup.py.\lucidrains\electra-pytorch\tests\test_electra_pytorch.py.\lucidrains\ema-pytorch\ema_pytorch\ema_pytorch.py.\lucidrains\ema-pytorch\ema_pytorch\post_hoc_ema.py.\lucidrains\ema-pytorch\ema_pytorch\__init__.py- EMA - Pytorch
.\lucidrains\ema-pytorch\setup.py.\lucidrains\En-transformer\denoise.py.\lucidrains\En-transformer\en_transformer\en_transformer.py.\lucidrains\En-transformer\en_transformer\utils.py.\lucidrains\En-transformer\en_transformer\__init__.py- E(n)-Equivariant Transformer
.\lucidrains\En-transformer\setup.py.\lucidrains\En-transformer\tests\test_equivariance.py.\lucidrains\enformer-pytorch\enformer_pytorch\config_enformer.py.\lucidrains\enformer-pytorch\enformer_pytorch\data.py.\lucidrains\enformer-pytorch\enformer_pytorch\finetune.py.\lucidrains\enformer-pytorch\enformer_pytorch\metrics.py.\lucidrains\enformer-pytorch\enformer_pytorch\modeling_enformer.py.\lucidrains\enformer-pytorch\enformer_pytorch\__init__.py.\lucidrains\enformer-pytorch\scripts\tf_to_torch.py.\lucidrains\enformer-pytorch\setup.py.\lucidrains\enformer-pytorch\test_pretrained.py.\lucidrains\enformer-tensorflow-sonnet-training-script\create_tfrecords.py- Enformer TPU training script (wip)
.\lucidrains\enformer-tensorflow-sonnet-training-script\sequence.py.\lucidrains\enformer-tensorflow-sonnet-training-script\train.py- Equiformer Diffusion (wip)
.\lucidrains\equiformer-pytorch\denoise.py.\lucidrains\equiformer-pytorch\equiformer_pytorch\basis.py.\lucidrains\equiformer-pytorch\equiformer_pytorch\equiformer_pytorch.py.\lucidrains\equiformer-pytorch\equiformer_pytorch\irr_repr.py.\lucidrains\equiformer-pytorch\equiformer_pytorch\reversible.py.\lucidrains\equiformer-pytorch\equiformer_pytorch\utils.py.\lucidrains\equiformer-pytorch\equiformer_pytorch\version.py.\lucidrains\equiformer-pytorch\equiformer_pytorch\__init__.py.\lucidrains\equiformer-pytorch\setup.py.\lucidrains\equiformer-pytorch\tests\test_edges.py.\lucidrains\equiformer-pytorch\tests\test_equivariance.py.\lucidrains\ESBN-pytorch\esbn_pytorch\esbn_pytorch.py.\lucidrains\ESBN-pytorch\esbn_pytorch\__init__.py.\lucidrains\ESBN-pytorch\setup.py.\lucidrains\esbn-transformer\esbn_transformer\esbn_transformer.py.\lucidrains\esbn-transformer\esbn_transformer\__init__.py- ESBN Transformer (wip)
.\lucidrains\ETSformer-pytorch\etsformer_pytorch\etsformer_pytorch.py.\lucidrains\ETSformer-pytorch\etsformer_pytorch\__init__.py.\lucidrains\ETSformer-pytorch\setup.py.\lucidrains\evolutionary-design-molecules\evolutionary_design_molecules\evolutionary_design_molecules.py.\lucidrains\evolutionary-design-molecules\evolutionary_design_molecules\__init__.py.\lucidrains\evolutionary-design-molecules\setup.py.\lucidrains\fast-transformer-pytorch\fast_transformer_pytorch\fast_transformer_pytorch.py.\lucidrains\fast-transformer-pytorch\fast_transformer_pytorch\__init__.py.\lucidrains\fast-transformer-pytorch\setup.py.\lucidrains\feedback-transformer-pytorch\feedback_transformer_pytorch\feedback_transformer_pytorch.py.\lucidrains\feedback-transformer-pytorch\feedback_transformer_pytorch\__init__.py- Feedback Transformer - Pytorch
.\lucidrains\feedback-transformer-pytorch\setup.py.\lucidrains\flamingo-pytorch\flamingo_pytorch\flamingo_palm.py.\lucidrains\flamingo-pytorch\flamingo_pytorch\flamingo_pytorch.py.\lucidrains\flamingo-pytorch\flamingo_pytorch\__init__.py.\lucidrains\flamingo-pytorch\setup.py.\lucidrains\flash-attention-jax\flash_attention_jax\attention.py.\lucidrains\flash-attention-jax\flash_attention_jax\causal_flash_attention.py.\lucidrains\flash-attention-jax\flash_attention_jax\cosine_sim_flash_attention.py.\lucidrains\flash-attention-jax\flash_attention_jax\flash_attention.py.\lucidrains\flash-attention-jax\flash_attention_jax\rabe_attention.py.\lucidrains\flash-attention-jax\flash_attention_jax\utils.py.\lucidrains\flash-attention-jax\flash_attention_jax\__init__.py.\lucidrains\flash-attention-jax\setup.py.\lucidrains\flash-cosine-sim-attention\benchmark.py- Data source
.\lucidrains\flash-cosine-sim-attention\flash_cosine_sim_attention\benchmark.py.\lucidrains\flash-cosine-sim-attention\flash_cosine_sim_attention\dispatch.h.\lucidrains\flash-cosine-sim-attention\flash_cosine_sim_attention\flash_cosine_sim_attention.py.\lucidrains\flash-cosine-sim-attention\flash_cosine_sim_attention\transformer.py.\lucidrains\flash-cosine-sim-attention\flash_cosine_sim_attention\version.py.\lucidrains\flash-cosine-sim-attention\flash_cosine_sim_attention\__init__.py.\lucidrains\flash-cosine-sim-attention\setup.py.\lucidrains\flash-cosine-sim-attention\tests\test.py.\lucidrains\flash-cosine-sim-attention\tests\__init__.py.\lucidrains\flash-cosine-sim-attention\train.py.\lucidrains\flash-genomics-model\flash_genomics_model\attend.py.\lucidrains\flash-genomics-model\flash_genomics_model\flash_genomics_model.py.\lucidrains\flash-genomics-model\flash_genomics_model\__init__.py- Flash Genomics Model (FGM)
.\lucidrains\flash-genomics-model\setup.py- Data source
.\lucidrains\FLASH-pytorch\flash_pytorch\autoregressive_wrapper.py.\lucidrains\FLASH-pytorch\flash_pytorch\flash_pytorch.py.\lucidrains\FLASH-pytorch\flash_pytorch\__init__.py.\lucidrains\FLASH-pytorch\setup.py.\lucidrains\FLASH-pytorch\train.py- Data source
.\lucidrains\g-mlp-gpt\g_mlp_gpt\autoregressive_wrapper.py.\lucidrains\g-mlp-gpt\g_mlp_gpt\g_mlp_gpt.py.\lucidrains\g-mlp-gpt\g_mlp_gpt\reversible.py.\lucidrains\g-mlp-gpt\g_mlp_gpt\__init__.py- GPT - gMLP
.\lucidrains\g-mlp-gpt\setup.py.\lucidrains\g-mlp-gpt\train.py- Data source
.\lucidrains\g-mlp-pytorch\g_mlp_pytorch\autoregressive_wrapper.py.\lucidrains\g-mlp-pytorch\g_mlp_pytorch\g_mlp_pytorch.py.\lucidrains\g-mlp-pytorch\g_mlp_pytorch\__init__.py.\lucidrains\g-mlp-pytorch\setup.py.\lucidrains\g-mlp-pytorch\train.py- Data source
.\lucidrains\gated-state-spaces-pytorch\gated_state_spaces_pytorch\autoregressive_wrapper.py.\lucidrains\gated-state-spaces-pytorch\gated_state_spaces_pytorch\dsconv.py.\lucidrains\gated-state-spaces-pytorch\gated_state_spaces_pytorch\gss.py.\lucidrains\gated-state-spaces-pytorch\gated_state_spaces_pytorch\mhesa.py.\lucidrains\gated-state-spaces-pytorch\gated_state_spaces_pytorch\__init__.py.\lucidrains\gated-state-spaces-pytorch\setup.py.\lucidrains\gated-state-spaces-pytorch\train.py- Data source
.\lucidrains\gateloop-transformer\gateloop_transformer\associative_scan.py.\lucidrains\gateloop-transformer\gateloop_transformer\gateloop_transformer.py.\lucidrains\gateloop-transformer\gateloop_transformer\gateloop_transformer_jax.py.\lucidrains\gateloop-transformer\gateloop_transformer\simplified_gate_loop.py.\lucidrains\gateloop-transformer\gateloop_transformer\__init__.py.\lucidrains\gateloop-transformer\setup.py.\lucidrains\gateloop-transformer\train.py.\lucidrains\genetic-algorithm-pytorch\bcga.py.\lucidrains\genetic-algorithm-pytorch\bega.py.\lucidrains\genetic-algorithm-pytorch\ga.py.\lucidrains\genetic-algorithm-pytorch\inbreed.py.\lucidrains\genetic-algorithm-pytorch\qbmb.py- genetic-algorithm-pytorch
.\lucidrains\geometric-vector-perceptron\examples\data_handler.py.\lucidrains\geometric-vector-perceptron\examples\data_utils.py- GVP - Point Cloud
.\lucidrains\geometric-vector-perceptron\examples\scn_data_module.py.\lucidrains\geometric-vector-perceptron\examples\train_lightning.py.\lucidrains\geometric-vector-perceptron\examples\__init__.py.\lucidrains\geometric-vector-perceptron\geometric_vector_perceptron\geometric_vector_perceptron.py.\lucidrains\geometric-vector-perceptron\geometric_vector_perceptron\__init__.py.\lucidrains\geometric-vector-perceptron\setup.py.\lucidrains\geometric-vector-perceptron\tests\tests.py.\lucidrains\gigagan-pytorch\gigagan_pytorch\attend.py.\lucidrains\gigagan-pytorch\gigagan_pytorch\data.py.\lucidrains\gigagan-pytorch\gigagan_pytorch\distributed.py.\lucidrains\gigagan-pytorch\gigagan_pytorch\gigagan_pytorch.py.\lucidrains\gigagan-pytorch\gigagan_pytorch\open_clip.py.\lucidrains\gigagan-pytorch\gigagan_pytorch\optimizer.py.\lucidrains\gigagan-pytorch\gigagan_pytorch\unet_upsampler.py.\lucidrains\gigagan-pytorch\gigagan_pytorch\version.py.\lucidrains\gigagan-pytorch\gigagan_pytorch\__init__.py.\lucidrains\gigagan-pytorch\setup.py.\lucidrains\global-self-attention-network\gsa_pytorch\gsa_pytorch.py.\lucidrains\global-self-attention-network\gsa_pytorch\__init__.py.\lucidrains\global-self-attention-network\setup.py.\lucidrains\glom-pytorch\glom_pytorch\glom_pytorch.py.\lucidrains\glom-pytorch\glom_pytorch\__init__.py.\lucidrains\glom-pytorch\setup.py.\lucidrains\gradnorm-pytorch\gradnorm_pytorch\gradnorm_pytorch.py.\lucidrains\gradnorm-pytorch\gradnorm_pytorch\mocks.py.\lucidrains\gradnorm-pytorch\gradnorm_pytorch\__init__.py.\lucidrains\gradnorm-pytorch\setup.py.\lucidrains\graph-transformer-pytorch\graph_transformer_pytorch\graph_transformer_pytorch.py.\lucidrains\graph-transformer-pytorch\graph_transformer_pytorch\__init__.py.\lucidrains\graph-transformer-pytorch\setup.py- Data source
.\lucidrains\h-transformer-1d\h_transformer_1d\autoregressive_wrapper.py.\lucidrains\h-transformer-1d\h_transformer_1d\h_transformer_1d.py.\lucidrains\h-transformer-1d\h_transformer_1d\reversible.py.\lucidrains\h-transformer-1d\h_transformer_1d\__init__.py.\lucidrains\h-transformer-1d\setup.py.\lucidrains\h-transformer-1d\train.py.\lucidrains\halonet-pytorch\halonet_pytorch\halonet_pytorch.py.\lucidrains\halonet-pytorch\halonet_pytorch\__init__.py.\lucidrains\halonet-pytorch\setup.py.\lucidrains\hamburger-pytorch\hamburger_pytorch\hamburger_pytorch.py.\lucidrains\hamburger-pytorch\hamburger_pytorch\__init__.py.\lucidrains\hamburger-pytorch\setup.py.\lucidrains\holodeck-pytorch\holodeck_pytorch\holodeck_pytorch.py.\lucidrains\holodeck-pytorch\holodeck_pytorch\__init__.py- Holodeck - Pytorch (wip)
.\lucidrains\holodeck-pytorch\setup.py- Data source
.\lucidrains\hourglass-transformer-pytorch\hourglass_transformer_pytorch\autoregressive_wrapper.py.\lucidrains\hourglass-transformer-pytorch\hourglass_transformer_pytorch\hourglass_transformer_pytorch.py.\lucidrains\hourglass-transformer-pytorch\hourglass_transformer_pytorch\__init__.py.\lucidrains\hourglass-transformer-pytorch\setup.py.\lucidrains\hourglass-transformer-pytorch\train.py.\lucidrains\HTM-pytorch\htm_pytorch\htm_pytorch.py.\lucidrains\HTM-pytorch\htm_pytorch\__init__.py.\lucidrains\HTM-pytorch\setup.py.\lucidrains\imagen-pytorch\imagen_pytorch\cli.py.\lucidrains\imagen-pytorch\imagen_pytorch\configs.py.\lucidrains\imagen-pytorch\imagen_pytorch\data.py.\lucidrains\imagen-pytorch\imagen_pytorch\elucidated_imagen.py.\lucidrains\imagen-pytorch\imagen_pytorch\imagen_pytorch.py.\lucidrains\imagen-pytorch\imagen_pytorch\imagen_video.py.\lucidrains\imagen-pytorch\imagen_pytorch\t5.py.\lucidrains\imagen-pytorch\imagen_pytorch\test\test_trainer.py.\lucidrains\imagen-pytorch\imagen_pytorch\test\__init__.py.\lucidrains\imagen-pytorch\imagen_pytorch\trainer.py.\lucidrains\imagen-pytorch\imagen_pytorch\utils.py.\lucidrains\imagen-pytorch\imagen_pytorch\version.py.\lucidrains\imagen-pytorch\imagen_pytorch\__init__.py.\lucidrains\imagen-pytorch\setup.py- Insertion Deletion Denoising Diffusion Probabilistic Models (wip)
.\lucidrains\invariant-point-attention\denoise.py.\lucidrains\invariant-point-attention\invariant_point_attention\invariant_point_attention.py.\lucidrains\invariant-point-attention\invariant_point_attention\utils.py.\lucidrains\invariant-point-attention\invariant_point_attention\__init__.py.\lucidrains\invariant-point-attention\setup.py.\lucidrains\invariant-point-attention\tests\invariance.py.\lucidrains\isab-pytorch\isab_pytorch\isab_pytorch.py.\lucidrains\isab-pytorch\isab_pytorch\__init__.py.\lucidrains\isab-pytorch\setup.py.\lucidrains\iTransformer\iTransformer\attend.py.\lucidrains\iTransformer\iTransformer\iTransformer.py.\lucidrains\iTransformer\iTransformer\iTransformer2D.py.\lucidrains\iTransformer\iTransformer\iTransformerFFT.py.\lucidrains\iTransformer\iTransformer\revin.py.\lucidrains\iTransformer\iTransformer\__init__.py.\lucidrains\iTransformer\setup.py.\lucidrains\ITTR-pytorch\ITTR_pytorch\ITTR_pytorch.py.\lucidrains\ITTR-pytorch\ITTR_pytorch\__init__.py.\lucidrains\ITTR-pytorch\setup.py.\lucidrains\jax2torch\jax2torch\jax2torch.py.\lucidrains\jax2torch\jax2torch\__init__.py- jax2torch
.\lucidrains\jax2torch\setup.py- Kalman Filtering Attention (wip)
.\lucidrains\kronecker-attention-pytorch\kronecker_attention_pytorch\kronecker_attention_pytorch.py.\lucidrains\kronecker-attention-pytorch\kronecker_attention_pytorch\__init__.py.\lucidrains\kronecker-attention-pytorch\setup.py.\lucidrains\lambda-networks\lambda_networks\lambda_networks.py.\lucidrains\lambda-networks\lambda_networks\tfkeras.py.\lucidrains\lambda-networks\lambda_networks\__init__.py.\lucidrains\lambda-networks\setup.py.\lucidrains\learning-to-expire-pytorch\learning_to_expire_pytorch\learning_to_expire_pytorch.py.\lucidrains\learning-to-expire-pytorch\learning_to_expire_pytorch\__init__.py.\lucidrains\learning-to-expire-pytorch\setup.py.\lucidrains\lie-transformer-pytorch\lie_transformer_pytorch\lie_transformer_pytorch.py.\lucidrains\lie-transformer-pytorch\lie_transformer_pytorch\reversible.py.\lucidrains\lie-transformer-pytorch\lie_transformer_pytorch\se3.py.\lucidrains\lie-transformer-pytorch\lie_transformer_pytorch\__init__.py.\lucidrains\lie-transformer-pytorch\setup.py.\lucidrains\lie-transformer-pytorch\tests.py.\lucidrains\lightweight-gan\lightweight_gan\cli.py.\lucidrains\lightweight-gan\lightweight_gan\diff_augment.py.\lucidrains\lightweight-gan\lightweight_gan\diff_augment_test.py.\lucidrains\lightweight-gan\lightweight_gan\lightweight_gan.py.\lucidrains\lightweight-gan\lightweight_gan\version.py.\lucidrains\lightweight-gan\lightweight_gan\__init__.py.\lucidrains\lightweight-gan\setup.py- Data source
- Linear Attention Transformer with Deepspeed for Enwik8
.\lucidrains\linear-attention-transformer\examples\enwik8_deepspeed\train.py- Data source
.\lucidrains\linear-attention-transformer\examples\enwik8_simple\train.py.\lucidrains\linear-attention-transformer\examples\toy_tasks\copy_task.py.\lucidrains\linear-attention-transformer\linear_attention_transformer\autopadder.py.\lucidrains\linear-attention-transformer\linear_attention_transformer\autoregressive_wrapper.py.\lucidrains\linear-attention-transformer\linear_attention_transformer\images.py.\lucidrains\linear-attention-transformer\linear_attention_transformer\linear_attention_transformer.py.\lucidrains\linear-attention-transformer\linear_attention_transformer\reversible.py.\lucidrains\linear-attention-transformer\linear_attention_transformer\__init__.py- Linear Attention Transformer
.\lucidrains\linear-attention-transformer\setup.py.\lucidrains\linformer\linformer\linformer.py.\lucidrains\linformer\linformer\reversible.py.\lucidrains\linformer\linformer\__init__.py- Linformer for Pytorch
.\lucidrains\linformer\setup.py.\lucidrains\lion-pytorch\lion_pytorch\lion_pytorch.py.\lucidrains\lion-pytorch\lion_pytorch\triton.py.\lucidrains\lion-pytorch\lion_pytorch\__init__.py.\lucidrains\lion-pytorch\setup.py- Liquid Conway Game of Life
.\lucidrains\liquid-conway\src\app.js.\lucidrains\liquid-conway\src\helpers.js.\lucidrains\liquid-conway\src\liquid.js.\lucidrains\liquid-conway\webpack.config.js.\lucidrains\llama-qrlhf\llama_qrlhf\llama.py.\lucidrains\llama-qrlhf\llama_qrlhf\llama_qrlhf.py.\lucidrains\llama-qrlhf\llama_qrlhf\__init__.py- Llama - QRLHF (wip)
.\lucidrains\llama-qrlhf\setup.py- Llama2 - Nim (wip)
- Data source
.\lucidrains\local-attention\local_attention\local_attention.py.\lucidrains\local-attention\local_attention\rotary.py.\lucidrains\local-attention\local_attention\transformer.py.\lucidrains\local-attention\local_attention\__init__.py.\lucidrains\local-attention\setup.py.\lucidrains\local-attention\train.py.\lucidrains\local-attention-flax\local_attention_flax\local_attention_flax.py.\lucidrains\local-attention-flax\local_attention_flax\__init__.py- Local Attention - Flax
.\lucidrains\local-attention-flax\setup.py.\lucidrains\logavgexp-torch\logavgexp_pytorch\logavgexp_pytorch.py.\lucidrains\logavgexp-torch\logavgexp_pytorch\__init__.py- LogAvgExp - Pytorch
.\lucidrains\logavgexp-torch\setup.py- Data source
.\lucidrains\long-short-transformer\long_short_transformer\autoregressive_wrapper.py.\lucidrains\long-short-transformer\long_short_transformer\long_short_transformer.py.\lucidrains\long-short-transformer\long_short_transformer\__init__.py.\lucidrains\long-short-transformer\setup.py.\lucidrains\long-short-transformer\train.py.\lucidrains\lumiere-pytorch\lumiere_pytorch\lumiere.py.\lucidrains\lumiere-pytorch\lumiere_pytorch\mp_lumiere.py.\lucidrains\lumiere-pytorch\lumiere_pytorch\__init__.py.\lucidrains\lumiere-pytorch\setup.py.\lucidrains\magvit2-pytorch\magvit2_pytorch\attend.py.\lucidrains\magvit2-pytorch\magvit2_pytorch\data.py.\lucidrains\magvit2-pytorch\magvit2_pytorch\magvit2_pytorch.py.\lucidrains\magvit2-pytorch\magvit2_pytorch\optimizer.py.\lucidrains\magvit2-pytorch\magvit2_pytorch\trainer.py.\lucidrains\magvit2-pytorch\magvit2_pytorch\version.py.\lucidrains\magvit2-pytorch\magvit2_pytorch\__init__.py.\lucidrains\magvit2-pytorch\setup.py.\lucidrains\make-a-video-pytorch\make_a_video_pytorch\attend.py.\lucidrains\make-a-video-pytorch\make_a_video_pytorch\make_a_video.py.\lucidrains\make-a-video-pytorch\make_a_video_pytorch\__init__.py.\lucidrains\make-a-video-pytorch\setup.py.\lucidrains\MaMMUT-pytorch\mammut_pytorch\mammut_pytorch.py.\lucidrains\MaMMUT-pytorch\mammut_pytorch\__init__.py.\lucidrains\MaMMUT-pytorch\setup.py.\lucidrains\marge-pytorch\marge_pytorch\autoregressive_wrapper.py.\lucidrains\marge-pytorch\marge_pytorch\marge_pytorch.py.\lucidrains\marge-pytorch\marge_pytorch\__init__.py.\lucidrains\marge-pytorch\setup.py.\lucidrains\med-seg-diff-pytorch\driver.py.\lucidrains\med-seg-diff-pytorch\med_seg_diff_pytorch\dataset.py.\lucidrains\med-seg-diff-pytorch\med_seg_diff_pytorch\med_seg_diff_pytorch.py.\lucidrains\med-seg-diff-pytorch\med_seg_diff_pytorch\__init__.py.\lucidrains\med-seg-diff-pytorch\sample.py.\lucidrains\med-seg-diff-pytorch\setup.py- Medical AI Experiments (wip)
.\lucidrains\medical-chatgpt\medical_chatgpt\medical_chatgpt.py.\lucidrains\medical-chatgpt\medical_chatgpt\__init__.py.\lucidrains\medical-chatgpt\setup.py- Data source
.\lucidrains\Mega-pytorch\mega_pytorch\autoregressive_wrapper.py.\lucidrains\Mega-pytorch\mega_pytorch\mega_pytorch.py.\lucidrains\Mega-pytorch\mega_pytorch\__init__.py.\lucidrains\Mega-pytorch\setup.py.\lucidrains\Mega-pytorch\train.py- Data source
.\lucidrains\MEGABYTE-pytorch\MEGABYTE_pytorch\attend.py.\lucidrains\MEGABYTE-pytorch\MEGABYTE_pytorch\megabyte.py.\lucidrains\MEGABYTE-pytorch\MEGABYTE_pytorch\__init__.py.\lucidrains\MEGABYTE-pytorch\setup.py.\lucidrains\MEGABYTE-pytorch\train.py.\lucidrains\memformer\memformer\autoregressive_wrapper.py.\lucidrains\memformer\memformer\memformer.py.\lucidrains\memformer\memformer\mrbp.py.\lucidrains\memformer\memformer\__init__.py.\lucidrains\memformer\setup.py- Data source
.\lucidrains\memorizing-transformers-pytorch\memorizing_transformers_pytorch\knn_memory.py.\lucidrains\memorizing-transformers-pytorch\memorizing_transformers_pytorch\memorizing_transformers_pytorch.py.\lucidrains\memorizing-transformers-pytorch\memorizing_transformers_pytorch\__init__.py.\lucidrains\memorizing-transformers-pytorch\setup.py.\lucidrains\memorizing-transformers-pytorch\train.py.\lucidrains\memory-compressed-attention\memory_compressed_attention\memory_compressed_attention.py.\lucidrains\memory-compressed-attention\memory_compressed_attention\__init__.py.\lucidrains\memory-compressed-attention\setup.py- Data source
.\lucidrains\memory-efficient-attention-pytorch\memory_efficient_attention_pytorch\autoregressive_wrapper.py.\lucidrains\memory-efficient-attention-pytorch\memory_efficient_attention_pytorch\cosine_sim_flash_attention.py.\lucidrains\memory-efficient-attention-pytorch\memory_efficient_attention_pytorch\flash_attention.py.\lucidrains\memory-efficient-attention-pytorch\memory_efficient_attention_pytorch\memory_efficient_attention.py.\lucidrains\memory-efficient-attention-pytorch\memory_efficient_attention_pytorch\memory_efficient_cosine_sim_attention.py.\lucidrains\memory-efficient-attention-pytorch\memory_efficient_attention_pytorch\reversible.py.\lucidrains\memory-efficient-attention-pytorch\memory_efficient_attention_pytorch\transformer.py.\lucidrains\memory-efficient-attention-pytorch\memory_efficient_attention_pytorch\__init__.py- Memory Efficient Attention Pytorch (obsolete)
.\lucidrains\memory-efficient-attention-pytorch\setup.py.\lucidrains\memory-efficient-attention-pytorch\tests\test.py.\lucidrains\memory-efficient-attention-pytorch\train.py- Data source
.\lucidrains\memory-transformer-xl\examples\enwik8_simple\train.py.\lucidrains\memory-transformer-xl\memory_transformer_xl\autoregressive_wrapper.py.\lucidrains\memory-transformer-xl\memory_transformer_xl\memory_transformer_xl.py.\lucidrains\memory-transformer-xl\memory_transformer_xl\__init__.py- Memory Transformer-XL
.\lucidrains\memory-transformer-xl\setup.py.\lucidrains\meshgpt-pytorch\meshgpt_pytorch\data.py.\lucidrains\meshgpt-pytorch\meshgpt_pytorch\meshgpt_pytorch.py.\lucidrains\meshgpt-pytorch\meshgpt_pytorch\trainer.py.\lucidrains\meshgpt-pytorch\meshgpt_pytorch\version.py.\lucidrains\meshgpt-pytorch\meshgpt_pytorch\__init__.py.\lucidrains\meshgpt-pytorch\setup.py- Data source
.\lucidrains\metaformer-gpt\metaformer_gpt\autoregressive_wrapper.py.\lucidrains\metaformer-gpt\metaformer_gpt\metaformer_gpt.py.\lucidrains\metaformer-gpt\metaformer_gpt\__init__.py.\lucidrains\metaformer-gpt\setup.py.\lucidrains\metaformer-gpt\train.py.\lucidrains\metnet3-pytorch\metnet3_pytorch\metnet3_pytorch.py.\lucidrains\metnet3-pytorch\metnet3_pytorch\__init__.py.\lucidrains\metnet3-pytorch\setup.py.\lucidrains\mirasol-pytorch\mirasol_pytorch\distributed.py.\lucidrains\mirasol-pytorch\mirasol_pytorch\mirasol_pytorch.py.\lucidrains\mirasol-pytorch\mirasol_pytorch\__init__.py.\lucidrains\mirasol-pytorch\setup.py- Data source
.\lucidrains\mixture-of-attention\mixture_of_attention\attend.py.\lucidrains\mixture-of-attention\mixture_of_attention\autoregressive_wrapper.py.\lucidrains\mixture-of-attention\mixture_of_attention\mixture_of_attention.py.\lucidrains\mixture-of-attention\mixture_of_attention\transformer.py.\lucidrains\mixture-of-attention\mixture_of_attention\__init__.py- Mixture-of-Attention
.\lucidrains\mixture-of-attention\setup.py.\lucidrains\mixture-of-attention\train.py.\lucidrains\mixture-of-experts\mixture_of_experts\mixture_of_experts.py.\lucidrains\mixture-of-experts\mixture_of_experts\__init__.py.\lucidrains\mixture-of-experts\setup.py.\lucidrains\mlm-pytorch\mlm_pytorch\mlm_pytorch.py.\lucidrains\mlm-pytorch\mlm_pytorch\__init__.py- MLM (Masked Language Modeling) Pytorch
.\lucidrains\mlm-pytorch\setup.py- Data source
.\lucidrains\mlp-gpt-jax\mlp_gpt_jax\mlp_gpt_jax.py.\lucidrains\mlp-gpt-jax\mlp_gpt_jax\utils.py.\lucidrains\mlp-gpt-jax\mlp_gpt_jax\__init__.py- MLP GPT - Jax
.\lucidrains\mlp-gpt-jax\setup.py.\lucidrains\mlp-gpt-jax\train.py.\lucidrains\mlp-mixer-pytorch\mlp_mixer_pytorch\mlp_mixer_pytorch.py.\lucidrains\mlp-mixer-pytorch\mlp_mixer_pytorch\permutator.py.\lucidrains\mlp-mixer-pytorch\mlp_mixer_pytorch\__init__.py.\lucidrains\mlp-mixer-pytorch\setup.py.\lucidrains\mogrifier\mogrifier\mogrifier.py.\lucidrains\mogrifier\mogrifier\__init__.py.\lucidrains\mogrifier\setup.py.\lucidrains\molecule-attention-transformer\molecule_attention_transformer\molecule_attention_transformer.py.\lucidrains\molecule-attention-transformer\molecule_attention_transformer\__init__.py.\lucidrains\molecule-attention-transformer\setup.py- Data source
.\lucidrains\multistream-transformers\multistream_transformers\autoregressive_wrapper.py.\lucidrains\multistream-transformers\multistream_transformers\multistream_transformers.py.\lucidrains\multistream-transformers\multistream_transformers\__init__.py.\lucidrains\multistream-transformers\setup.py.\lucidrains\multistream-transformers\train.py.\lucidrains\muse-maskgit-pytorch\muse_maskgit_pytorch\attend.py.\lucidrains\muse-maskgit-pytorch\muse_maskgit_pytorch\muse_maskgit_pytorch.py.\lucidrains\muse-maskgit-pytorch\muse_maskgit_pytorch\t5.py.\lucidrains\muse-maskgit-pytorch\muse_maskgit_pytorch\trainers.py.\lucidrains\muse-maskgit-pytorch\muse_maskgit_pytorch\vqgan_vae.py.\lucidrains\muse-maskgit-pytorch\muse_maskgit_pytorch\__init__.py.\lucidrains\muse-maskgit-pytorch\setup.py.\lucidrains\musiclm-pytorch\musiclm_pytorch\distributed.py.\lucidrains\musiclm-pytorch\musiclm_pytorch\musiclm_pytorch.py.\lucidrains\musiclm-pytorch\musiclm_pytorch\trainer.py.\lucidrains\musiclm-pytorch\musiclm_pytorch\__init__.py.\lucidrains\musiclm-pytorch\setup.py.\lucidrains\n-grammer-pytorch\n_grammer_pytorch\n_grammer_pytorch.py.\lucidrains\n-grammer-pytorch\n_grammer_pytorch\__init__.py.\lucidrains\n-grammer-pytorch\setup.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\aligner.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\attend.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\naturalspeech2_pytorch.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\cleaner.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\expand\abbreviations.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\expand\number_norm.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\expand\time_norm.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\expand\__init__.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\phonemizers\base.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\phonemizers\espeak_wrapper.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\phonemizers\punctuation.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\phonemizers\__init__.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\tokenizer.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\utils.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\utils\__init__.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\version.py.\lucidrains\naturalspeech2-pytorch\naturalspeech2_pytorch\__init__.py.\lucidrains\naturalspeech2-pytorch\setup.py.\lucidrains\neural-plexer-pytorch\neural_plexer_pytorch\neural_plexer_pytorch.py.\lucidrains\neural-plexer-pytorch\neural_plexer_pytorch\__init__.py- NeuralPlexer - Pytorch (wip)
.\lucidrains\neural-plexer-pytorch\setup.py- Neural Sequence Chunkers - Pytorch (wip)
- nim-genetic-algorithm
- Nim Tokenizer (wip)
.\lucidrains\nuwa-pytorch\nuwa_pytorch\image_utils.py.\lucidrains\nuwa-pytorch\nuwa_pytorch\nuwa_pytorch.py.\lucidrains\nuwa-pytorch\nuwa_pytorch\optimizer.py.\lucidrains\nuwa-pytorch\nuwa_pytorch\reversible.py.\lucidrains\nuwa-pytorch\nuwa_pytorch\reversible_video_audio.py.\lucidrains\nuwa-pytorch\nuwa_pytorch\train_nuwa.py.\lucidrains\nuwa-pytorch\nuwa_pytorch\train_vqgan_vae.py.\lucidrains\nuwa-pytorch\nuwa_pytorch\vqgan_vae.py.\lucidrains\nuwa-pytorch\nuwa_pytorch\__init__.py.\lucidrains\nuwa-pytorch\setup.py.\lucidrains\NWT-pytorch\nwt_pytorch\nwt_pytorch.py.\lucidrains\NWT-pytorch\nwt_pytorch\__init__.py- NWT - Pytorch (wip)
.\lucidrains\NWT-pytorch\setup.py.\lucidrains\nystrom-attention\nystrom_attention\nystrom_attention.py.\lucidrains\nystrom-attention\nystrom_attention\__init__.py.\lucidrains\nystrom-attention\setup.py.\lucidrains\omninet-pytorch\omninet_pytorch\omninet_pytorch.py.\lucidrains\omninet-pytorch\omninet_pytorch\__init__.py.\lucidrains\omninet-pytorch\setup.py- Data source
.\lucidrains\PaLM-jax\palm_jax\palm.py.\lucidrains\PaLM-jax\palm_jax\palm_lite.py.\lucidrains\PaLM-jax\palm_jax\utils.py.\lucidrains\PaLM-jax\palm_jax\__init__.py.\lucidrains\PaLM-jax\setup.py.\lucidrains\PaLM-jax\train.py- Data source
- PaLM-pytorch with Deepspeed for Enwik8
.\lucidrains\PaLM-pytorch\examples\enwik8_deepspeed\train.py.\lucidrains\PaLM-pytorch\palm_pytorch\autoregressive_wrapper.py.\lucidrains\PaLM-pytorch\palm_pytorch\palm_lite.py.\lucidrains\PaLM-pytorch\palm_pytorch\palm_pytorch.py.\lucidrains\PaLM-pytorch\palm_pytorch\triton\layernorm.py.\lucidrains\PaLM-pytorch\palm_pytorch\triton\palm.py.\lucidrains\PaLM-pytorch\palm_pytorch\triton\softmax.py.\lucidrains\PaLM-pytorch\palm_pytorch\triton\__init__.py.\lucidrains\PaLM-pytorch\palm_pytorch\__init__.py.\lucidrains\PaLM-pytorch\setup.py.\lucidrains\PaLM-pytorch\train.py- Data source
.\lucidrains\PaLM-rlhf-pytorch\palm_rlhf_pytorch\attention.py.\lucidrains\PaLM-rlhf-pytorch\palm_rlhf_pytorch\lora.py.\lucidrains\PaLM-rlhf-pytorch\palm_rlhf_pytorch\optimizer.py.\lucidrains\PaLM-rlhf-pytorch\palm_rlhf_pytorch\palm.py.\lucidrains\PaLM-rlhf-pytorch\palm_rlhf_pytorch\ppo.py.\lucidrains\PaLM-rlhf-pytorch\palm_rlhf_pytorch\reward.py.\lucidrains\PaLM-rlhf-pytorch\palm_rlhf_pytorch\utils.py.\lucidrains\PaLM-rlhf-pytorch\palm_rlhf_pytorch\__init__.py.\lucidrains\PaLM-rlhf-pytorch\setup.py.\lucidrains\PaLM-rlhf-pytorch\train.py.\lucidrains\panoptic-transformer\panoptic_transformer\data.py.\lucidrains\panoptic-transformer\panoptic_transformer\panoptic_transformer.py.\lucidrains\panoptic-transformer\panoptic_transformer\__init__.py- Panoptic Transformer (wip)
.\lucidrains\panoptic-transformer\scripts\gen-pathx.py.\lucidrains\panoptic-transformer\setup.py.\lucidrains\parti-pytorch\parti_pytorch\optimizer.py.\lucidrains\parti-pytorch\parti_pytorch\parti_pytorch.py.\lucidrains\parti-pytorch\parti_pytorch\t5.py.\lucidrains\parti-pytorch\parti_pytorch\version.py.\lucidrains\parti-pytorch\parti_pytorch\vit_vqgan.py.\lucidrains\parti-pytorch\parti_pytorch\vit_vqgan_trainer.py.\lucidrains\parti-pytorch\parti_pytorch\__init__.py.\lucidrains\parti-pytorch\setup.py.\lucidrains\pause-transformer\pause_transformer\pause_transformer.py.\lucidrains\pause-transformer\pause_transformer\__init__.py.\lucidrains\pause-transformer\setup.py- Data source
.\lucidrains\perceiver-ar-pytorch\perceiver_ar_pytorch\autoregressive_wrapper.py.\lucidrains\perceiver-ar-pytorch\perceiver_ar_pytorch\perceiver_ar_pytorch.py.\lucidrains\perceiver-ar-pytorch\perceiver_ar_pytorch\__init__.py.\lucidrains\perceiver-ar-pytorch\setup.py.\lucidrains\perceiver-ar-pytorch\train.py.\lucidrains\perceiver-pytorch\perceiver_pytorch\experimental.py.\lucidrains\perceiver-pytorch\perceiver_pytorch\gated.py.\lucidrains\perceiver-pytorch\perceiver_pytorch\mixed_latents.py.\lucidrains\perceiver-pytorch\perceiver_pytorch\perceiver_io.py.\lucidrains\perceiver-pytorch\perceiver_pytorch\perceiver_pytorch.py.\lucidrains\perceiver-pytorch\perceiver_pytorch\__init__.py.\lucidrains\perceiver-pytorch\setup.py- Data source
- Performer with Deepspeed for Enwik8
.\lucidrains\performer-pytorch\examples\enwik8_deepspeed\train.py- Data source
.\lucidrains\performer-pytorch\examples\enwik8_simple\train.py.\lucidrains\performer-pytorch\examples\toy_tasks\enc_dec_copy.py.\lucidrains\performer-pytorch\examples\toy_tasks\enc_dec_copy_apex.py.\lucidrains\performer-pytorch\performer_pytorch\autoregressive_wrapper.py.\lucidrains\performer-pytorch\performer_pytorch\performer_enc_dec.py.\lucidrains\performer-pytorch\performer_pytorch\performer_pytorch.py.\lucidrains\performer-pytorch\performer_pytorch\reversible.py.\lucidrains\performer-pytorch\performer_pytorch\__init__.py.\lucidrains\performer-pytorch\setup.py.\lucidrains\perfusion-pytorch\perfusion_pytorch\embedding.py.\lucidrains\perfusion-pytorch\perfusion_pytorch\open_clip.py.\lucidrains\perfusion-pytorch\perfusion_pytorch\optimizer.py.\lucidrains\perfusion-pytorch\perfusion_pytorch\perfusion.py.\lucidrains\perfusion-pytorch\perfusion_pytorch\save_load.py.\lucidrains\perfusion-pytorch\perfusion_pytorch\__init__.py.\lucidrains\perfusion-pytorch\setup.py.\lucidrains\phasic-policy-gradient\train.py.\lucidrains\phenaki-pytorch\phenaki_pytorch\attention.py.\lucidrains\phenaki-pytorch\phenaki_pytorch\cvivit.py.\lucidrains\phenaki-pytorch\phenaki_pytorch\cvivit_trainer.py.\lucidrains\phenaki-pytorch\phenaki_pytorch\data.py.\lucidrains\phenaki-pytorch\phenaki_pytorch\optimizer.py.\lucidrains\phenaki-pytorch\phenaki_pytorch\phenaki_pytorch.py.\lucidrains\phenaki-pytorch\phenaki_pytorch\phenaki_trainer.py.\lucidrains\phenaki-pytorch\phenaki_pytorch\t5.py.\lucidrains\phenaki-pytorch\phenaki_pytorch\__init__.py.\lucidrains\phenaki-pytorch\setup.py.\lucidrains\pi-GAN-pytorch\pi_gan_pytorch\coordconv.py.\lucidrains\pi-GAN-pytorch\pi_gan_pytorch\nerf.py.\lucidrains\pi-GAN-pytorch\pi_gan_pytorch\pi_gan_pytorch.py.\lucidrains\pi-GAN-pytorch\pi_gan_pytorch\__init__.py.\lucidrains\pi-GAN-pytorch\setup.py.\lucidrains\pixel-level-contrastive-learning\pixel_level_contrastive_learning\pixel_level_contrastive_learning.py.\lucidrains\pixel-level-contrastive-learning\pixel_level_contrastive_learning\__init__.py.\lucidrains\pixel-level-contrastive-learning\setup.py.\lucidrains\point-transformer-pytorch\point_transformer_pytorch\multihead_point_transformer_pytorch.py.\lucidrains\point-transformer-pytorch\point_transformer_pytorch\point_transformer_pytorch.py.\lucidrains\point-transformer-pytorch\point_transformer_pytorch\__init__.py.\lucidrains\point-transformer-pytorch\setup.py.\lucidrains\ponder-transformer\ponder_transformer\ponder_transformer.py.\lucidrains\ponder-transformer\ponder_transformer\__init__.py- Ponder(ing) Transformer
.\lucidrains\ponder-transformer\setup.py- Data source
.\lucidrains\product-key-memory\product_key_memory\product_key_memory.py.\lucidrains\product-key-memory\product_key_memory\transformer.py.\lucidrains\product-key-memory\product_key_memory\__init__.py.\lucidrains\product-key-memory\setup.py.\lucidrains\product-key-memory\train.py.\lucidrains\progen\generate_data.py.\lucidrains\progen\progen_transformer\checkpoint.py.\lucidrains\progen\progen_transformer\data.py.\lucidrains\progen\progen_transformer\progen.py.\lucidrains\progen\progen_transformer\utils.py.\lucidrains\progen\progen_transformer\__init__.py- ProGen - (wip)
.\lucidrains\progen\sample.py.\lucidrains\progen\train.py.\lucidrains\protein-bert-pytorch\protein_bert_pytorch\protein_bert_pytorch.py.\lucidrains\protein-bert-pytorch\protein_bert_pytorch\__init__.py- ProteinBERT - Pytorch (wip)
.\lucidrains\protein-bert-pytorch\setup.py.\lucidrains\pytorch-custom-utils\pytorch_custom_utils\accelerate_utils.py.\lucidrains\pytorch-custom-utils\pytorch_custom_utils\get_adam_optimizer.py.\lucidrains\pytorch-custom-utils\pytorch_custom_utils\module_device.py.\lucidrains\pytorch-custom-utils\pytorch_custom_utils\optimizer_scheduler_warmup.py.\lucidrains\pytorch-custom-utils\pytorch_custom_utils\save_load.py.\lucidrains\pytorch-custom-utils\pytorch_custom_utils\total_parameters.py.\lucidrains\pytorch-custom-utils\pytorch_custom_utils\utils.py.\lucidrains\pytorch-custom-utils\pytorch_custom_utils\__init__.py- Pytorch Custom Utils (wip)
.\lucidrains\pytorch-custom-utils\setup.py.\lucidrains\q-transformer\q_transformer\agent.py.\lucidrains\q-transformer\q_transformer\attend.py.\lucidrains\q-transformer\q_transformer\mocks.py.\lucidrains\q-transformer\q_transformer\optimizer.py.\lucidrains\q-transformer\q_transformer\q_learner.py.\lucidrains\q-transformer\q_transformer\q_robotic_transformer.py.\lucidrains\q-transformer\q_transformer\__init__.py.\lucidrains\q-transformer\setup.py.\lucidrains\quartic-transformer\quartic_transformer\multi_stream_transformer.py.\lucidrains\quartic-transformer\quartic_transformer\quartic_transformer.py.\lucidrains\quartic-transformer\quartic_transformer\__init__.py- Quartic Transformer (wip)
.\lucidrains\quartic-transformer\setup.py.\lucidrains\recurrent-interface-network-pytorch\rin_pytorch\attend.py.\lucidrains\recurrent-interface-network-pytorch\rin_pytorch\rin_pytorch.py.\lucidrains\recurrent-interface-network-pytorch\rin_pytorch\__init__.py.\lucidrains\recurrent-interface-network-pytorch\setup.py- Data source
.\lucidrains\recurrent-memory-transformer-pytorch\recurrent_memory_transformer_pytorch\attend.py.\lucidrains\recurrent-memory-transformer-pytorch\recurrent_memory_transformer_pytorch\recurrent_memory_transformer.py.\lucidrains\recurrent-memory-transformer-pytorch\recurrent_memory_transformer_pytorch\__init__.py.\lucidrains\recurrent-memory-transformer-pytorch\setup.py.\lucidrains\recurrent-memory-transformer-pytorch\train.py- Data source
- Reformer with Deepspeed for Enwik8
.\lucidrains\reformer-pytorch\examples\enwik8_deepspeed\train.py- Data source
.\lucidrains\reformer-pytorch\examples\enwik8_simple\train.py- Pretraining
.\lucidrains\reformer-pytorch\pretraining\self-supervised.py- Reformer, the Efficient Transformer, in Pytorch
.\lucidrains\reformer-pytorch\reformer_pytorch\autopadder.py.\lucidrains\reformer-pytorch\reformer_pytorch\generative_tools.py.\lucidrains\reformer-pytorch\reformer_pytorch\recorder.py.\lucidrains\reformer-pytorch\reformer_pytorch\reformer_enc_dec.py.\lucidrains\reformer-pytorch\reformer_pytorch\reformer_pytorch.py.\lucidrains\reformer-pytorch\reformer_pytorch\reversible.py.\lucidrains\reformer-pytorch\reformer_pytorch\__init__.py.\lucidrains\reformer-pytorch\setup.py- Data source
.\lucidrains\rela-transformer\rela_transformer\autoregressive_wrapper.py.\lucidrains\rela-transformer\rela_transformer\rela_transformer.py.\lucidrains\rela-transformer\rela_transformer\__init__.py.\lucidrains\rela-transformer\setup.py.\lucidrains\rela-transformer\train.py.\lucidrains\remixer-pytorch\remixer_pytorch\remixer_pytorch.py.\lucidrains\remixer-pytorch\remixer_pytorch\__init__.py.\lucidrains\remixer-pytorch\setup.py.\lucidrains\res-mlp-pytorch\res_mlp_pytorch\res_mlp_pytorch.py.\lucidrains\res-mlp-pytorch\res_mlp_pytorch\__init__.py.\lucidrains\res-mlp-pytorch\setup.py- Retrieval-Augmented Denoising Diffusion Probabilistic Models (wip)
.\lucidrains\retrieval-augmented-ddpm\retrieval_augmented_ddpm\retrieval_augmented_ddpm.py.\lucidrains\retrieval-augmented-ddpm\retrieval_augmented_ddpm\__init__.py.\lucidrains\retrieval-augmented-ddpm\setup.py.\lucidrains\RETRO-pytorch\retro_pytorch\data.py.\lucidrains\RETRO-pytorch\retro_pytorch\optimizer.py.\lucidrains\RETRO-pytorch\retro_pytorch\retrieval.py.\lucidrains\RETRO-pytorch\retro_pytorch\retro_pytorch.py.\lucidrains\RETRO-pytorch\retro_pytorch\training.py.\lucidrains\RETRO-pytorch\retro_pytorch\utils.py.\lucidrains\RETRO-pytorch\retro_pytorch\__init__.py.\lucidrains\RETRO-pytorch\setup.py.\lucidrains\ring-attention-pytorch\assert.py.\lucidrains\ring-attention-pytorch\assert_flash.py.\lucidrains\ring-attention-pytorch\ring_attention_pytorch\distributed.py.\lucidrains\ring-attention-pytorch\ring_attention_pytorch\ring.py.\lucidrains\ring-attention-pytorch\ring_attention_pytorch\ring_attention.py.\lucidrains\ring-attention-pytorch\ring_attention_pytorch\ring_flash_attention.py.\lucidrains\ring-attention-pytorch\ring_attention_pytorch\ring_flash_attention_cuda.py.\lucidrains\ring-attention-pytorch\ring_attention_pytorch\__init__.py.\lucidrains\ring-attention-pytorch\setup.py.\lucidrains\robotic-transformer-pytorch\robotic_transformer_pytorch\robotic_transformer_pytorch.py.\lucidrains\robotic-transformer-pytorch\robotic_transformer_pytorch\__init__.py.\lucidrains\robotic-transformer-pytorch\setup.py.\lucidrains\rotary-embedding-torch\rotary_embedding_torch\rotary_embedding_torch.py.\lucidrains\rotary-embedding-torch\rotary_embedding_torch\__init__.py.\lucidrains\rotary-embedding-torch\setup.py- Data source
- Routing Transformer with Deepspeed for Enwik8
.\lucidrains\routing-transformer\examples\enwik8_deepspeed\train.py- Data source
.\lucidrains\routing-transformer\examples\enwik8_simple\train.py.\lucidrains\routing-transformer\examples\toy_tasks\enc_dec_copy_task.py.\lucidrains\routing-transformer\examples\toy_tasks\increment.py- Routing Transformer
.\lucidrains\routing-transformer\routing_transformer\autopadder.py.\lucidrains\routing-transformer\routing_transformer\autoregressive_wrapper.py.\lucidrains\routing-transformer\routing_transformer\encoder_decoder.py.\lucidrains\routing-transformer\routing_transformer\reversible.py.\lucidrains\routing-transformer\routing_transformer\routing_transformer.py.\lucidrains\routing-transformer\routing_transformer\__init__.py.\lucidrains\routing-transformer\setup.py- Data source
.\lucidrains\RQ-Transformer\rq_transformer\hierarchical_causal_transformer.py.\lucidrains\RQ-Transformer\rq_transformer\rq_transformer.py.\lucidrains\RQ-Transformer\rq_transformer\__init__.py.\lucidrains\RQ-Transformer\setup.py.\lucidrains\RQ-Transformer\train.py- Data source
- RVQ-VAE-GPT - Residual Vector Quantize VAE - GPT (wip)
.\lucidrains\rvq-vae-gpt\rvq_vae_gpt\rvq_vae_gpt.py.\lucidrains\rvq-vae-gpt\rvq_vae_gpt\__init__.py.\lucidrains\rvq-vae-gpt\setup.py.\lucidrains\rvq-vae-gpt\train.py- SAC (Soft Actor Critic) - Pytorch (wip)
.\lucidrains\SAC-pytorch\SAC_pytorch\SAC.py.\lucidrains\SAC-pytorch\SAC_pytorch\__init__.py.\lucidrains\SAC-pytorch\setup.py.\lucidrains\scattering-compositional-learner\scattering_transform\scattering_transform.py.\lucidrains\scattering-compositional-learner\scattering_transform\__init__.py.\lucidrains\scattering-compositional-learner\setup.py.\lucidrains\se3-transformer-pytorch\denoise.py.\lucidrains\se3-transformer-pytorch\se3_transformer_pytorch\basis.py.\lucidrains\se3-transformer-pytorch\se3_transformer_pytorch\irr_repr.py.\lucidrains\se3-transformer-pytorch\se3_transformer_pytorch\reversible.py.\lucidrains\se3-transformer-pytorch\se3_transformer_pytorch\rotary.py.\lucidrains\se3-transformer-pytorch\se3_transformer_pytorch\se3_transformer_pytorch.py.\lucidrains\se3-transformer-pytorch\se3_transformer_pytorch\spherical_harmonics.py.\lucidrains\se3-transformer-pytorch\se3_transformer_pytorch\utils.py.\lucidrains\se3-transformer-pytorch\se3_transformer_pytorch\__init__.py.\lucidrains\se3-transformer-pytorch\setup.py.\lucidrains\se3-transformer-pytorch\tests\test_basis.py.\lucidrains\se3-transformer-pytorch\tests\test_equivariance.py.\lucidrains\se3-transformer-pytorch\tests\test_irrep_repr.py.\lucidrains\se3-transformer-pytorch\tests\test_spherical_harmonics.py.\lucidrains\segformer-pytorch\segformer_pytorch\segformer_pytorch.py.\lucidrains\segformer-pytorch\segformer_pytorch\__init__.py.\lucidrains\segformer-pytorch\setup.py.\lucidrains\self-rewarding-lm-pytorch\self_rewarding_lm_pytorch\dpo.py.\lucidrains\self-rewarding-lm-pytorch\self_rewarding_lm_pytorch\mocks.py.\lucidrains\self-rewarding-lm-pytorch\self_rewarding_lm_pytorch\sampling_utils.py.\lucidrains\self-rewarding-lm-pytorch\self_rewarding_lm_pytorch\self_rewarding_lm_pytorch.py.\lucidrains\self-rewarding-lm-pytorch\self_rewarding_lm_pytorch\spin.py.\lucidrains\self-rewarding-lm-pytorch\self_rewarding_lm_pytorch\__init__.py.\lucidrains\self-rewarding-lm-pytorch\setup.py- Data source
- Simple Hierarchical Transformer
.\lucidrains\simple-hierarchical-transformer\setup.py.\lucidrains\simple-hierarchical-transformer\simple_hierarchical_transformer\attention.py.\lucidrains\simple-hierarchical-transformer\simple_hierarchical_transformer\simple_hierarchical_transformer.py.\lucidrains\simple-hierarchical-transformer\simple_hierarchical_transformer\__init__.py.\lucidrains\simple-hierarchical-transformer\train.py- Data source
- Sinkhorn Transformer with Deepspeed for Enwik8
.\lucidrains\sinkhorn-transformer\examples\enwik8_deepspeed\train.py- Data source
.\lucidrains\sinkhorn-transformer\examples\enwik8_simple\train.py.\lucidrains\sinkhorn-transformer\examples\increment_by_one\train.py- Sinkhorn Transformer
.\lucidrains\sinkhorn-transformer\setup.py.\lucidrains\sinkhorn-transformer\sinkhorn_transformer\autopadder.py.\lucidrains\sinkhorn-transformer\sinkhorn_transformer\autoregressive_wrapper.py.\lucidrains\sinkhorn-transformer\sinkhorn_transformer\reversible.py.\lucidrains\sinkhorn-transformer\sinkhorn_transformer\sinkhorn_transformer.py.\lucidrains\sinkhorn-transformer\sinkhorn_transformer\__init__.py- SIREN in Pytorch
.\lucidrains\siren-pytorch\setup.py.\lucidrains\siren-pytorch\siren_pytorch\siren_pytorch.py.\lucidrains\siren-pytorch\siren_pytorch\__init__.py.\lucidrains\slot-attention\setup.py.\lucidrains\slot-attention\slot_attention\slot_attention.py.\lucidrains\slot-attention\slot_attention\slot_attention_experimental.py.\lucidrains\slot-attention\slot_attention\__init__.py.\lucidrains\soft-moe-pytorch\assert.py.\lucidrains\soft-moe-pytorch\setup.py.\lucidrains\soft-moe-pytorch\soft_moe_pytorch\distributed.py.\lucidrains\soft-moe-pytorch\soft_moe_pytorch\soft_moe.py.\lucidrains\soft-moe-pytorch\soft_moe_pytorch\soft_moe_with_dynamic_slots.py.\lucidrains\soft-moe-pytorch\soft_moe_pytorch\__init__.py.\lucidrains\soundstorm-pytorch\setup.py.\lucidrains\soundstorm-pytorch\soundstorm_pytorch\attend.py.\lucidrains\soundstorm-pytorch\soundstorm_pytorch\soundstorm.py.\lucidrains\soundstorm-pytorch\soundstorm_pytorch\trainer.py.\lucidrains\soundstorm-pytorch\soundstorm_pytorch\__init__.py.\lucidrains\spear-tts-pytorch\setup.py.\lucidrains\spear-tts-pytorch\spear_tts_pytorch\attend.py.\lucidrains\spear-tts-pytorch\spear_tts_pytorch\data.py.\lucidrains\spear-tts-pytorch\spear_tts_pytorch\distributed.py.\lucidrains\spear-tts-pytorch\spear_tts_pytorch\spear_tts_pytorch.py.\lucidrains\spear-tts-pytorch\spear_tts_pytorch\trainer.py.\lucidrains\spear-tts-pytorch\spear_tts_pytorch\__init__.py- Data source
.\lucidrains\speculative-decoding\setup.py.\lucidrains\speculative-decoding\speculative_decoding\speculative_decoding.py.\lucidrains\speculative-decoding\speculative_decoding\speculative_decoding_with_prophet.py.\lucidrains\speculative-decoding\speculative_decoding\__init__.py.\lucidrains\speculative-decoding\train.py.\lucidrains\speculative-decoding\train_early_exit.py.\lucidrains\speculative-decoding\train_prophet.py.\lucidrains\st-moe-pytorch\assert.py.\lucidrains\st-moe-pytorch\setup.py.\lucidrains\st-moe-pytorch\st_moe_pytorch\distributed.py.\lucidrains\st-moe-pytorch\st_moe_pytorch\st_moe_pytorch.py.\lucidrains\st-moe-pytorch\st_moe_pytorch\__init__.py.\lucidrains\STAM-pytorch\setup.py.\lucidrains\STAM-pytorch\stam_pytorch\stam.py.\lucidrains\STAM-pytorch\stam_pytorch\__init__.py- Simple StyleGan2 for Pytorch
.\lucidrains\stylegan2-pytorch\setup.py.\lucidrains\stylegan2-pytorch\stylegan2_pytorch\cli.py.\lucidrains\stylegan2-pytorch\stylegan2_pytorch\diff_augment.py.\lucidrains\stylegan2-pytorch\stylegan2_pytorch\stylegan2_pytorch.py.\lucidrains\stylegan2-pytorch\stylegan2_pytorch\version.py.\lucidrains\stylegan2-pytorch\stylegan2_pytorch\__init__.py.\lucidrains\tab-transformer-pytorch\setup.py.\lucidrains\tab-transformer-pytorch\tab_transformer_pytorch\ft_transformer.py.\lucidrains\tab-transformer-pytorch\tab_transformer_pytorch\tab_transformer_pytorch.py.\lucidrains\tab-transformer-pytorch\tab_transformer_pytorch\__init__.py.\lucidrains\taylor-series-linear-attention\setup.py.\lucidrains\taylor-series-linear-attention\taylor_series_linear_attention\attention.py.\lucidrains\taylor-series-linear-attention\taylor_series_linear_attention\vit.py.\lucidrains\taylor-series-linear-attention\taylor_series_linear_attention\__init__.py.\lucidrains\tf-bind-transformer\finetune_binary_pred.py.\lucidrains\tf-bind-transformer\finetune_track.py.\lucidrains\tf-bind-transformer\precache_proteins.py- Transcription Factor binding predictions with Attention and Transformers
.\lucidrains\tf-bind-transformer\scripts\download_experiments.py.\lucidrains\tf-bind-transformer\scripts\fetch_factor_fastas.py.\lucidrains\tf-bind-transformer\scripts\negative_peak_to_bool_npy.py.\lucidrains\tf-bind-transformer\scripts\remap_to_separate_exp_target_cell_beds.py.\lucidrains\tf-bind-transformer\setup.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\attention.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\cache_utils.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\context_utils.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\data.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\data_bigwig.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\gene_utils.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\optimizer.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\protein_utils.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\tf_bind_transformer.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\training_utils.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\training_utils_bigwig.py.\lucidrains\tf-bind-transformer\tf_bind_transformer\__init__.py.\lucidrains\TimeSformer-pytorch\setup.py.\lucidrains\TimeSformer-pytorch\timesformer_pytorch\rotary.py.\lucidrains\TimeSformer-pytorch\timesformer_pytorch\timesformer_pytorch.py.\lucidrains\TimeSformer-pytorch\timesformer_pytorch\__init__.py- Data source
- Token Shift GPT
.\lucidrains\token-shift-gpt\setup.py.\lucidrains\token-shift-gpt\token_shift_gpt\autoregressive_wrapper.py.\lucidrains\token-shift-gpt\token_shift_gpt\token_shift_gpt.py.\lucidrains\token-shift-gpt\token_shift_gpt\__init__.py.\lucidrains\token-shift-gpt\train.py.\lucidrains\toolformer-pytorch\setup.py.\lucidrains\toolformer-pytorch\toolformer_pytorch\optimizer.py.\lucidrains\toolformer-pytorch\toolformer_pytorch\palm.py.\lucidrains\toolformer-pytorch\toolformer_pytorch\prompts.py.\lucidrains\toolformer-pytorch\toolformer_pytorch\toolformer_pytorch.py.\lucidrains\toolformer-pytorch\toolformer_pytorch\tools.py.\lucidrains\toolformer-pytorch\toolformer_pytorch\__init__.py- TPDNE (wip)
.\lucidrains\TPDNE\setup.py.\lucidrains\TPDNE\TPDNE_utils\tpdne.py.\lucidrains\TPDNE\TPDNE_utils\__init__.py.\lucidrains\tr-rosetta-pytorch\setup.py.\lucidrains\tr-rosetta-pytorch\tr_rosetta_pytorch\cli.py.\lucidrains\tr-rosetta-pytorch\tr_rosetta_pytorch\tr_rosetta_pytorch.py.\lucidrains\tr-rosetta-pytorch\tr_rosetta_pytorch\utils.py.\lucidrains\tr-rosetta-pytorch\tr_rosetta_pytorch\__init__.py.\lucidrains\tranception-pytorch\setup.py.\lucidrains\tranception-pytorch\tranception_pytorch\tranception_pytorch.py.\lucidrains\tranception-pytorch\tranception_pytorch\__init__.py.\lucidrains\transformer-in-transformer\setup.py.\lucidrains\transformer-in-transformer\transformer_in_transformer\tnt.py.\lucidrains\transformer-in-transformer\transformer_in_transformer\__init__.py.\lucidrains\transframer-pytorch\setup.py.\lucidrains\transframer-pytorch\transframer_pytorch\transframer_pytorch.py.\lucidrains\transframer-pytorch\transframer_pytorch\__init__.py- TransGanFormer (wip)
.\lucidrains\transganformer\setup.py.\lucidrains\transganformer\transganformer\cli.py.\lucidrains\transganformer\transganformer\diff_augment.py.\lucidrains\transganformer\transganformer\transganformer.py.\lucidrains\transganformer\transganformer\version.py.\lucidrains\transganformer\transganformer\__init__.py.\lucidrains\triangle-multiplicative-module\setup.py.\lucidrains\triangle-multiplicative-module\triangle_multiplicative_module\triangle_multiplicative_module.py.\lucidrains\triangle-multiplicative-module\triangle_multiplicative_module\__init__.py.\lucidrains\triton-transformer\assert.py- Data source
- Transformer in Triton (wip)
.\lucidrains\triton-transformer\setup.py.\lucidrains\triton-transformer\train.py.\lucidrains\triton-transformer\triton_transformer\autoregressive_wrapper.py.\lucidrains\triton-transformer\triton_transformer\bmm.py.\lucidrains\triton-transformer\triton_transformer\cross_entropy.py.\lucidrains\triton-transformer\triton_transformer\dropout.py.\lucidrains\triton-transformer\triton_transformer\layernorm.py.\lucidrains\triton-transformer\triton_transformer\softmax.py.\lucidrains\triton-transformer\triton_transformer\transformer.py.\lucidrains\triton-transformer\triton_transformer\utils.py.\lucidrains\triton-transformer\triton_transformer\__init__.py.\lucidrains\uformer-pytorch\setup.py.\lucidrains\uformer-pytorch\uformer_pytorch\uformer_pytorch.py.\lucidrains\uformer-pytorch\uformer_pytorch\__init__.py.\lucidrains\unet-stylegan2\setup.py.\lucidrains\unet-stylegan2\unet_stylegan2\diff_augment.py.\lucidrains\unet-stylegan2\unet_stylegan2\unet_stylegan2.py.\lucidrains\unet-stylegan2\unet_stylegan2\__init__.py.\lucidrains\uniformer-pytorch\setup.py.\lucidrains\uniformer-pytorch\uniformer_pytorch\uniformer_pytorch.py.\lucidrains\uniformer-pytorch\uniformer_pytorch\__init__.py.\lucidrains\vector-quantize-pytorch\examples\autoencoder.py.\lucidrains\vector-quantize-pytorch\examples\autoencoder_fsq.py.\lucidrains\vector-quantize-pytorch\examples\autoencoder_lfq.py.\lucidrains\vector-quantize-pytorch\setup.py.\lucidrains\vector-quantize-pytorch\vector_quantize_pytorch\finite_scalar_quantization.py.\lucidrains\vector-quantize-pytorch\vector_quantize_pytorch\latent_quantization.py.\lucidrains\vector-quantize-pytorch\vector_quantize_pytorch\lookup_free_quantization.py.\lucidrains\vector-quantize-pytorch\vector_quantize_pytorch\random_projection_quantizer.py.\lucidrains\vector-quantize-pytorch\vector_quantize_pytorch\residual_fsq.py.\lucidrains\vector-quantize-pytorch\vector_quantize_pytorch\residual_lfq.py.\lucidrains\vector-quantize-pytorch\vector_quantize_pytorch\residual_vq.py.\lucidrains\vector-quantize-pytorch\vector_quantize_pytorch\vector_quantize_pytorch.py.\lucidrains\vector-quantize-pytorch\vector_quantize_pytorch\__init__.py.\lucidrains\video-diffusion-pytorch\setup.py.\lucidrains\video-diffusion-pytorch\video_diffusion_pytorch\text.py.\lucidrains\video-diffusion-pytorch\video_diffusion_pytorch\video_diffusion_pytorch.py.\lucidrains\video-diffusion-pytorch\video_diffusion_pytorch\__init__.py.\lucidrains\vit-pytorch\setup.py.\lucidrains\vit-pytorch\tests\test.py.\lucidrains\vit-pytorch\vit_pytorch\ats_vit.py.\lucidrains\vit-pytorch\vit_pytorch\cait.py.\lucidrains\vit-pytorch\vit_pytorch\cct.py.\lucidrains\vit-pytorch\vit_pytorch\cct_3d.py.\lucidrains\vit-pytorch\vit_pytorch\crossformer.py.\lucidrains\vit-pytorch\vit_pytorch\cross_vit.py.\lucidrains\vit-pytorch\vit_pytorch\cvt.py.\lucidrains\vit-pytorch\vit_pytorch\deepvit.py.\lucidrains\vit-pytorch\vit_pytorch\dino.py.\lucidrains\vit-pytorch\vit_pytorch\distill.py.\lucidrains\vit-pytorch\vit_pytorch\efficient.py.\lucidrains\vit-pytorch\vit_pytorch\es_vit.py.\lucidrains\vit-pytorch\vit_pytorch\extractor.py.\lucidrains\vit-pytorch\vit_pytorch\learnable_memory_vit.py.\lucidrains\vit-pytorch\vit_pytorch\levit.py.\lucidrains\vit-pytorch\vit_pytorch\local_vit.py.\lucidrains\vit-pytorch\vit_pytorch\mae.py.\lucidrains\vit-pytorch\vit_pytorch\max_vit.py.\lucidrains\vit-pytorch\vit_pytorch\max_vit_with_registers.py.\lucidrains\vit-pytorch\vit_pytorch\mobile_vit.py.\lucidrains\vit-pytorch\vit_pytorch\mp3.py.\lucidrains\vit-pytorch\vit_pytorch\mpp.py.\lucidrains\vit-pytorch\vit_pytorch\na_vit.py.\lucidrains\vit-pytorch\vit_pytorch\nest.py.\lucidrains\vit-pytorch\vit_pytorch\parallel_vit.py- [
.\lucidrains\vit-pytorch\vit_pytorch\pit.py](https://github.com/
腾讯云面向开发者汇聚海量精品云计算使用和开发经验，营造开放的云计算技术生态圈。
更多推荐
6
0- 0
已为社区贡献4条内容
所有评论(0)